数组中的第k个最大元素—leetcode215

在未排序的数组中找到第 k 个最大的元素。请注意，你需要找的是数组排序后的第 k 个最大的元素，而不是第 k 个不同的元素。

示例 1:

输入: [3,2,1,5,6,4] 和 k = 2
输出: 5
示例 2:

输入: [3,2,3,1,2,4,5,5,6] 和 k = 4
输出: 4
说明:

你可以假设 k 总是有效的，且 1 ≤ k ≤ 数组的长度。

 解法一：冒泡排序，从后往前排，复杂度O(kn)

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        int n = nums.size();
        if(n==0){
            return 0;
        }
        int start = 0;
        while(start<k){
            for(int i=n-1;i>start;--i){
                if(nums[i]>nums[i-1]){
                    int temp = nums[i-1];
                    nums[i-1] = nums[i];
                    nums[i] = temp;
                }
            }
            start++;
        }
        
        return nums[k-1];
    }
};

解法二：小顶堆，复杂度O(nlogk)

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        int N=nums.size();
        build_Ksmallstack(nums,k);
        for(int index=k;index<N;++index){
            if(nums[index]>nums[0]){
                swap(nums,index,0);
                smallstack(nums, k, 0);
            }
        }
        return nums[0];
    }

    void build_Ksmallstack(vector<int>& nums, int k){
        int noTreeNodes = (k-1)/2;
        for(int i=noTreeNodes;i>=0;--i){
            smallstack(nums,k,i);
        }
        return;
    }

    void smallstack(vector<int>& nums, int k, int index){
        int min = index;
        while(true){
            if(2*index+1<k && nums[2*index+1]<nums[index]){
                min = 2*index+1;
            }
            if(2*index+2<k && nums[2*index+2]<nums[min]){
                min = 2*index+2;
            }
            if(min==index){
                break;
            }
            swap(nums,min,index);
            index = min;
        }
    }

    void swap(vector<int>& nums,int min,int index){
        int temp = nums[min];
        nums[min] = nums[index];
        nums[index] = temp;
        return;
    }
};