LeetCode215分治法快速选择

在未排序的数组中找到第 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 ≤ 数组的长度。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/kth-largest-element-in-an-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

首先找出第K个最大元素,可以直接把数组排序,再从数组索引最后一个向内移动,直到第K个元素即可输出。这样做的话复杂度就直接是O(nlgn)的复杂度

快速选择:

类似快速排序的操作将前K大的元素都放在数组最右边 定义nums[0]等于i(数组起始为head,末尾为tail) 利用快速排序的双指针,将他置于某一中间位置,用数组的长度减去这时的位置即可得到大于i 元素的个数,使其与k比较,若大于k则说明包含位置靠前了,大于i的元素包含k了,则从i的位置+1 到tail再进行一次选择 反之则位置靠后,则从head到i的位置-1再进行一次选择,当数组长度减k=0的时候则说明所有比k大的元素都放在了k的后面数组该位置的元素即可

public class Solution {
        int nums[] ;
        public int findKthLargest(int[] nums, int k) {
            this.nums = nums ;
            quickSelect(0,nums.length-1,k);
            return nums[nums.length - k ];
        }
        //类似于快速排序的双指针
        void quickSelect( int head, int tail, int k ) {
            int i = this.nums[head];
            int x = head ;
            int y = tail ;
            while (x != y) {
                while (x < y && i <= this.nums[y]) {
                    y-- ;
                }
                swap(x,y);
                while (y > x && i >= this.nums[x]) {
                    x++;
                }
                swap(x,y);
            }
            //对于指针位置的判断,因为x=y的所以判断一个就可以了
            if (k == nums.length-x) {
                return;
            } else if (nums.length-x > k) {
                quickSelect(x+1,tail,k);
            } else {
                quickSelect(head,x-1,k);
            }
        }
        //交换
        void swap( int i, int j) {
            int temp = this.nums[i];
            this.nums[i] = this.nums[j];
            this.nums[j] = temp ;
        }
}

复杂度分析:平均情况O(N),最坏情况 O(N2)

想是这样写,写出来真的慢的离谱。。。还不如直接调库排序然后输出第K个元素快。。。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值