数组中的第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：

import java.util.Arrays;

class Solution {

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

 

解法2：

class Solution {

    public int findKthLargest(int[] nums, int k) {
        solution(nums, 0, nums.length - 1, k);
        return nums[k - 1];
    }

    private void solution(int[] nums, int begin, int end, int k){
    	if(begin >= end){
    		return;
    	}

    	int partition = partition(nums, begin, end);
    	if(partition == k - 1){
    		return;
    	}
    	if(partition < k - 1){
    		solution(nums, partition + 1, end, k);
    	}else{
    		solution(nums, begin, partition - 1, k);
    	}    	
    }

    private int partition(int[] nums, int begin, int end){
    	int left = nums[begin];
    	while(begin < end){
    		while(begin < end && nums[end] <= left){
    			end--;
    		}
    		nums[begin] = nums[end];
    		while(begin < end && nums[begin] >= left){
    			begin++;
    		}
    		nums[end] = nums[begin];
    	}

    	nums[begin] = left;
    	return begin;
    }
}