CongliYin的博客

达则兼济天下,穷则独善其身。

Leetcode#215. Kth Largest Element in an Array

题目描述:找出数组中第 k 大的数

解题思路:

  • 使用数组内容构建一个最大堆/最小堆,通过每次pop出堆顶后继续维护堆的结构,直到满足一定的次数(最大堆k-1次,最小堆size-k次),堆顶的元素就是第k大的数字;
  • 利用快排的partition函数思想,选定一个数组内的值作为pivot,将小于pivot的数字放到pivot右边,大于等于pivot的数字放到pivot左边。接着判断两边数字的数量,如果左边的数量小于k个,说明第k大的数字存在于pivot及pivot右边的区域之内,对右半区执行partition函数;如果右边的数量小于k个,说明第k大的数字在pivot和pivot左边的区域之内,对左半区执行partition函数。直到左半区刚好有k-1个数,那么第k大的数就已经找到了。
  • *
class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) 
    {
        //建一个最大堆
        make_heap(nums.begin(), nums.end());
        int ret;
        for(int i = 0; i < k; ++i)
        {
            ret = nums.front();
            //第一个元素被移到最后面,剩下的元素继续维持最大堆
            pop_heap(nums.begin(), nums.end());
            //将最后一个元素出堆
            nums.pop_back();
        }
        //以上执行了k步,被取出来的数就是第 k 大的数
        return ret;
    }
};
class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) 
    {
        int low = 0, high = nums.size() - 1;
        while(low <= high)
        {
            int i = low;
            int j = high;
            int pivot = nums[low];
            while(i <= j)
            {
                while(i <= j && nums[i] >= pivot)
                    i++;
                while(i <= j && nums[j] <= pivot)
                    j--;
                if(i < j)
                    swap(nums[i], nums[j]);
            }
            swap(nums[low], nums[j]);
            if(j == k - 1)
                return nums[j];
            else if(j < k - 1)
                low = j + 1;
            else
                high = j - 1;
        }
    }
};
阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/sinat_20177327/article/details/79972970
上一篇Leetcode#62. Unique Paths
下一篇Leetcode#19. Remove Nth Node From End of List
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭