Kth Largest Element in an Array

我的新博客地址：HuHaoyu’s Blog，欢迎参观。

Find the **k**th largest element in an unsorted array. Note that it is the kth largest element in the sorted order, not the kth distinct element.

For example,
Given [3,2,1,5,6,4] and k = 2, return 5.

Note:
You may assume k is always valid, 1 ≤ k ≤ array's length.

题意

找到一个无序数组中第k大的数。

分析

类似快速排序一样。每层递归中，对当前的nums区段内的元素进行操作：

• 保存当前区段的第一个元素nums[bottom]为save。
• 类似快排，将大于save的元素移到左侧，小于save的移到右侧。
• 通过下标相减得知save在当前区段中的次序rank。
• 如rank == k，则save就是第k大的数。
• 如rank < k，则第k大的数在save所在位置的右侧，进入下一层递归。
• 如rank > k，则第k大的数在save所在位置的左侧，进入下一层递归。

AC代码

class Solution {
public:
    int answer;
    int findKthLargest(vector<int>& nums, int k) {
        find(nums, 0, nums.size() - 1, k);
        return answer;
    }

    void find(vector<int>& nums, int bottom, int top, int k) {
        int i = bottom, j = top, save = nums[bottom];
        while (i < j) {
            for (; i < j && nums[j] <= save; --j);
            nums[i] = nums[j];
            for (; i < j && nums[i] >= save; ++i);
            nums[j] = nums[i];
        }

        int rank = j - bottom + 1;
        if (rank == k) {
            answer = save; return;
        }
        if (rank > k) {
            find(nums, bottom, j - 1, k);
        } else {
            find(nums, j + 1, top, k - rank);
        }
    }
};

如代码或分析有误，请批评指正，谢谢。