LeetCode Kth Largest Element in an Array

思路:

先排序后选择,时间复杂度:O(N*logN),空间复杂度:O(1)。

c++ code:

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        sort(nums.begin(), nums.end());
        return nums[nums.size() - k];
    }
};

利用快排的思想,用第一个元素作为pivot,以它为界分割其它所有元素,再看左部分的长度已决定下一次如何操作。
平均时间复杂度:O(N),空间复杂度:O(1)。

c++ code:

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        int L = 0, R = nums.size() - 1;
        while (L < R) {
            int left = L, right = R;
            int key = nums[left];
            while (left < right) {
                while (left < right && nums[right] < key) --right;
                nums[left] = nums[right];
                while (left < right && nums[left] >= key) ++left;
                nums[right] = nums[left];
            }
            nums[left] = key;
            if (left == k - 1) return nums[k - 1];
            else if (left > k - 1) R = left - 1;
            else L = left + 1;
        }
        return nums[k - 1];
    }
};

DISCUSS中利用堆的方法:

make_heap(RandomAccessIterator first, RandomAccessIterator last) 默认生成的是大顶推,可以使用 make_heap (RandomAccessIterator first, RandomAccessIterator last,Compare comp ) 自定义的cmp来生成小顶堆。

pop_heap() 首元素和末元素交换,再重新堆排序。
pop_back() 删除容器中最后一个元素。

c++ code:

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        make_heap(nums.begin(), nums.end());
        for(auto i=0; i<k-1;i++) {
            pop_heap(nums.begin(), nums.end());
            nums.pop_back();
        }
        return nums.front();
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值