0
点赞
收藏
分享

微信扫一扫

[模板总结] - 快速选择 (霍尔Quick Selection)

alanwhy 2022-01-31 阅读 57

模板题目 & 链接

Leetcode 215 Kth Largest Element 第K大的元素

Leetcode 347. Top K Frequent Elements TopK高频元素

基本思路:

快速选择是快速排序算法的衍生品,当对数组进行partition分组时,实际上就是将比pivot大或者小的分成一组,加入pivot数组在整个数组中是第K大的元素,那么partition后pivot之前就是前K-1大的元素集合,只是顺序不确定。根据这个快速排序partition的性质,可以引出快速选择的基本问询功能:

思路就是快速排序类似,代码如下:

private void quickSelect(int[] nums, int start, int end, int k) {
        
        if(start>end) return;
        
        int l = start, r = end;
        int mid = l + (r-l>>1);
        int pivot = nums[mid];
        
        while(l<=r) {
            while(l<=r && nums[l]<pivot) l++;
            while(l<=r && nums[r]>pivot) r--;
            
            if(l<=r) {
                int temp = nums[l];
                nums[l] = nums[r];
                nums[r] = temp;
                l++;
                r--;
            }
        }
        
        // k在前一半
        if(k<=r) quickSelect(nums, start, r, k);
        else quickSelect(nums, l, end, k);
                
    }

时间复杂度:O(N), 最坏情况:O(N^{2})每一次partition都只分割出一个元素的顺序(一组N-1个,一组1个);空间复杂度:O(1)。 

 

举报

相关推荐

0 条评论