0
点赞
收藏
分享

微信扫一扫

Leetcode215 第K大的元素

幸甚至哉歌以咏志 2022-03-14 阅读 69

        本题是快速排序的变形题目。题目描述如下:

        本题的基本思想如下:

        在快排每次划分区间的过程中,计算比枢轴元素大的元素个数sr,即sr = nums.size()-j-1. 然后将sr与k作对比:若k<=sr,则取右半边进行递归;否则取左半边进行递归。注意边界条件是left==right,此时返回nums[left].

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        int left = 0, right = nums.size()-1;
        int res = partition(nums,left,right,k);
        return res;      
    }
 
    int partition(vector<int>&nums,int left, int right,int k){
        if(left==right) return nums[left];
        int i=left-1, j=right+1, x=nums[left];
        while(i<j){
            do i++; while(nums[i]<x);
            do j--; while(nums[j]>x);
            if(i<j) swap(nums[i],nums[j]);
        }
        int sr = nums.size()-j-1;
        if(k<=sr) return partition(nums,j+1,right,k);
        else return partition(nums,left,j,k);
    }
};

        本算法的平均时间复杂度是O(n),而如果简单地使用快速排序的平均时间复杂度为O(n^2)

举报

相关推荐

0 条评论