leetcode学习笔记215 kth-largest-element-in-an-array

问题

kth-largest-element-in-an-array

给定一个数组,找到里面第K大的数.

Example 1:

Input: [3,2,1,5,6,4] and k = 2
Output: 5

Example 2:

Input: [3,2,3,1,2,4,5,5,6] and k = 4
Output: 4

方法1

最简单的方法就是先排列,然后找到第K大的数就可以了.

时间复杂度O(NlogN).
空间复杂度O(1).

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

方法2

稍微复杂一些的就是自定义快速排序,在排序的过程中寻找第K大的数. 而不是把整个数组排序完成.

时间复杂度O(N).
空间复杂度O(1).

class Solution {
    public int findKthLargest(int[] nums, int k) {
	    // 第k大, 也就是第 nums.length - k 小
        return helper(nums, nums.length - k, 0, nums.length - 1);
    }
    
    private int helper(int[] nums, int k, int left, int right){
        // 把最左边定义为比较基准,这个地方其实更好的方法是将left到right中随机的一个位置和left进行交换之后再进行计算
        // 因为我们也不知道要找的数字出现在哪个位置, 随便交换之后再进行排序,可以在大样本的情况下防止出现最差的情况
        // int randomIndex = left + new Random().nextInt(right - left + 1);
        // swamp(nums,left, randomIndex);
        int temp = nums[left];
        int index = right;
        for(int i = right; i > left; i--){
            if(nums[i] > temp)
                swamp(nums, i, index--);
        }
        swamp(nums, left, index);
        if(index < k)
            return helper(nums, k, index + 1, right);
        else if(index > k)
            return helper(nums, k, left, index - 1);
        else
            return nums[k];  
    }
    
    private void swamp(int[] nums, int i, int j){
        int temp = nums[i];
        nums[i] = nums[j];
        nums[j] = temp;
    }
}