Leetcode 第K大元素

题目:

  • 在数组中找到第 k 大的元素。
  • 要求时间复杂度为O(n),空间复杂度为O(1)。

输入:
n = 1, nums = [1,3,4,2]
输出:
4

分析:

最开始想到是利用快排序。快速排序在每一次swap之后都能够找到某一个元素(例如 a a a)正确的位置。则:

  • a a a刚好是第K个元素,则输出;
  • a a a的位置在第K个位置右侧,则接下来去其左侧快排;
  • a a a的位置在第K个位置左侧,则接下来去其右侧快排;

代码:

class Solution {
   
public:
    /**
     * @param n: An integer
     * @param nums: An array
     * @return: the Kth largest element
     */
    int kthLargestElement(int n, vector<int> &nums) {
   
        // write your code here
        return getRes(nums,n,0,nums.size()-1);
    }
    int getRes(vector<int> &nums,int k,int left, int right){
   
        int tmp=nums[left],n=nums.size(),tleft=left,tright=right;
        while(left<right){
   
            while(right>left&&nums[right]>tmp) right--;
            if(right>left){
   
                nums[left++]=nums[right];
            }
            while(left<right&&nums[left]<=tmp) left++;
            if(left<right) nums[right--]=nums[left];
        }
        nums[left]=tmp;
        if(left==
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值