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

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

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

思路

这里主要用到了快速排序的方法,将输入的 vector从大到小排序,然 后再在排好序的 vector中选第 k个就行了。 先定义了一个公共变量 orinums,所有的位移都在这里面进行; qiucksort函数做了一个快速排序:制造这个 vector的一个轴点(左边 全是大于这个点的数,右边全是小于这个点的数),轴点用 pivotsort 来计算得到,所谓制造就是确定一个点(随机选取会比较科学,可以 降低碰到反向的有序排列时去第一个点而增加计算量的概率)作为轴 点,然后将大于这个点的值放在左边,小于这个点的值放在右边。 Pivotsort函数主要想法就是:分别从左边右边扫描各个值,符合条件 的就继续扫描,不符合就将它赋值给另一边的位置,之所以把轴点 pivot定义出来就是为了留个空位用来赋值。

class Solution {
public:
    vector<int> orinums;
    int findKthLargest(vector<int>& nums, int k) {
        orinums = nums;
        int n = nums.size();
        if(n<2){return nums[0];}
        quicksort(0,n-1);
        return orinums[k-1];
    }
    void quicksort(int lo,int hi){
        if(hi-lo<1){return;}
        int mi = pivotsort(lo,hi);
        quicksort(lo,mi);
        quicksort(mi+1,hi);
    }
    int pivotsort(int lo,int hi){
        int pivot = orinums[lo];
        while(lo<hi){
        while((lo<hi)&&(orinums[hi]<=pivot)){
            hi--;
        }
        orinums[lo] = orinums[hi];
        while((lo<hi)&&(orinums[lo]>=pivot)){
            lo++;
        }
        orinums[hi]=orinums[lo];
        }
        orinums[lo]=pivot;
        return lo;

    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
可以使用快速选择算法来找出数组中第 k 个小的元素。 快速选择算法的基本思想是,选择一个基准元素,将数组中小于基准元素元素放在基准元素的左边,将大于基准元素元素放在基准元素的右边,然后比较基准元素的位置和 k 的大小关系,如果基准元素的位置等于 k,就找到了第 k 个小的元素,如果基准元素的位置大于 k,就在基准元素的左边继续查找,如果基准元素的位置小于 k,就在基准元素的右边继续查找。 以下是使用快速选择算法找出数组中第 k 个小的元素的 Python 代码: ```python def quick_select(nums, k): pivot = nums[0] left = [x for x in nums if x < pivot] right = [x for x in nums if x > pivot] mid = [x for x in nums if x == pivot] if k <= len(left): return quick_select(left, k) elif k <= len(left) + len(mid): return pivot else: return quick_select(right, k - len(left) - len(mid)) ``` 在这个函数,我们首先选择数组中的第一个元素作为基准元素 pivot。然后,我们将数组分成三个部分:小于 pivot 的元素放在 left 数组中,大于 pivot 的元素放在 right 数组中,等于 pivot 的元素放在 mid 数组中。接下来,我们比较基准元素的位置和 k 的大小关系,如果基准元素的位置等于 k,就返回基准元素的值,如果基准元素的位置大于 k,就在 left 数组中继续查找第 k 个小的元素,如果基准元素的位置小于 k,就在 right 数组中继续查找第 k - len(left) - len(mid) 个小的元素。 这个函数的时间复杂度为 O(n),其 n 是数组的长度。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值