第k大的数

排序的方法就不说了

这里记录一下o(n)复杂度的算法

思想,利用快速排序的划分思想,每一次划分会有一个元素处于最终位置

判断这个位置,若所处位置大于k,则向左边寻找

若所处位置大于k,向右边寻找。

第k大的数,是指数组从小到大排序完后右边往左数第k个,也就是左边往右数下标nums.size()-k 位置的数

class Solution {
public:
    int divide(vector<int> &nums, int left, int right){
        int middle = nums[left];
        while(left<right){
            while(left<right&&nums[right]>=middle) right--;
            nums[left]=nums[right];
            while(left<right&&nums[left]<=middle) left++;
            nums[right]=nums[left];
        }
        nums[left]=middle;
        return left;
    }

    int find_K(vector<int> &nums, int left, int right, int k){
        if(left==right){
            if(k==left)
                return nums[left];
            else
                return -1;
        }
        int c = divide(nums, left, right);
        if(k==c){
            return nums[c];
        }else if (k<c){
            return find_K(nums,left,c-1,k);
        }else{
            return find_K(nums,c+1,right,k);
        }
    }

    int findKthLargest(vector<int>& nums, int k) {
        int r = find_K(nums,0,nums.size()-1, nums.size()-k);
        return r;
    }
};

 

posted @ 2020-03-14 19:26  ecnu_lxz  阅读(308)  评论(0编辑  收藏  举报