leetcode215.数组中的第k个最大元素(快排/堆排序 java)***

class Solution {
    public int findKthLargest(int[] nums, int k) {
        int len=nums.length;
        int left=0,right=len-1;
        int target=len-k;                                //第k大的元素的数组下标就是len-k,可通过找规律得出
        while(true){
            int index=partition(nums,left,right);       //index用来判断nums[left]排在数组中的第几位大小,并将其移到对应的位置
            if(index==target)
                return nums[target];
            if(index>target)                         //因为nums[left]比第k大的数大,nums[left]左边的数都比它小,所以使right=index-1.继续接下来的判断
                right=index-1;
            if(index<target)
                left=index+1;
        }
    }

    public int partition(int[] nums, int left, int right) {
        int pivot=nums[left];
        int j=left;                                   //j用来判断pivot排在数组中的第几个大小位置
        for(int i=left+1;i<=right;i++){
            if(nums[i]<pivot){
                j++;
                swap(nums,j,i);
            }
        }
        swap(nums,j,left);
        return j;
    }

    private void swap(int[] nums, int index1, int index2) {
        if(index1==index2) return ;
        int temp=nums[index1];
        nums[index1]=nums[index2];
        nums[index2]=temp;
    }
}

猜你喜欢

转载自www.cnblogs.com/y1040511302/p/11455788.html