code - 数组中的第K个最大元素

题目来源:leetcode

链接: https://leetcode-cn.com/problems/kth-largest-element-in-an-array/

215. 数组中的第K个最大元素

难度中等483收藏分享切换为英文关注反馈

在未排序的数组中找到第 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 ≤ 数组的长度。

 

代码如下,利用快速排序的原理进行. 

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        //第k个最大,相当于第nums.size()-k个最小
        return quick(nums, nums.size()-k+1, 0, nums.size()-1);

    }

    int quick(vector<int>& nums, int k, int begin_index, int end_index)
    {
        if (begin_index == end_index && k == 1)
        {
            return nums[begin_index];
        }

        // 选择一个数作为临界值
        int index = (begin_index+end_index)/2;
        int tmp = nums[index];
        nums[index] = nums[begin_index]; 
        nums[begin_index] = tmp; 
        int i = begin_index;
        int j = end_index;
        
        //从小到大排序
        while (i < j)
        {
            while (nums[j] > tmp && j > i)
            {
                j--;
            }
            if (j > i)
            {
                nums[i] = nums[j];
                i++;
            }

            while (nums[i] <= tmp && j > i)
            {
                i++;
            }
            if (j > i)
            {
                nums[j] = nums[i];
                j--;
            }
            nums[i] = tmp;
        }

        if (i-begin_index >= k)
        {
            return quick(nums, k, begin_index, i-1);
        }
        else if (i-begin_index + 1 == k)
        {
            return nums[i];
        }
        else // i-begin_index + 1 < k
        {
            return quick(nums, k-(i-begin_index+1), i+1, end_index);
        }
        

    }
};

 结果:

执行用时 :12 ms, 在所有 C++ 提交中击败了91.44%的用户

内存消耗 :9.9 MB, 在所有 C++ 提交中击败了5.88%的用户

 

需要注意,在快速排序的时候, 如果将临界值选择第一个元素,耗时很大,换成中间元素就是上面的结果. leetcode的测试数据集感觉有点坑啊