leetcode 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 ≤ 数组的长度。

解法1:先排序再查找

public int findKthLargest(int[] nums, int k) {
    Arrays.sort(nums);
    return nums[nums.length - k];
}

解法2:使用快排的思想

       核心思想是每次都要先找一个中枢点Pivot,然后遍历其他所有的数字,像这道题从小往大排的话,就把小于中枢点的数字放到左半边,把大于中枢点的放在右半边,这样中枢点是整个数组中第几大的数字就确定了,虽然左右两部分不一定是完全有序的,但是并不影响本题要求的结果,所以我们求出中枢点的位置,如果正好是k-1,那么直接返回该位置上的数字;如果大于k-1,说明要求的数字在左半部分,更新右边界,再求新的中枢点位置;反之则更新右半部分,求中枢点的位置。

public int findKthLargest(int[] nums, int k) {
    int left = 0;
    int right = nums.length - 1;
    while(true){
        int pos = partation(nums, left, right);
        if(pos == k - 1) return nums[pos];
        if(pos > k - 1){
            right = pos - 1;
        }else{
            left = pos + 1;
        }
    }
}
public int partation(int[] nums, int left, int right){
    int privote = nums[left], l = left + 1, r = right;
    while(l <= r){
        if(nums[l] < privote && nums[r] > privote){    
            swap(nums, l++, r--);
        }
        if(nums[l] >= privote) l++;
        if(nums[r] <= privote) r--;
    }
    swap(nums, left, r);
    return r;
}
private void swap(int[] nums, int i, int j){
    int temp = nums[left];
    nums[left] = nums[right];
    nums[right] = temp;
}

猜你喜欢

转载自blog.csdn.net/qq_43322057/article/details/84783529
今日推荐