LeetCode 215 Kth Largest Element in an Array

问题描述:

解答:一开始我自己的思路,用一个长度为K的数组记录前K大的数字。

class Solution {
    public int findKthLargest(int[] nums, int k) {
        int[] temp=new int[k];
        Arrays.sort(nums,0,k);
        for(int i=0;i<k;i++) {
            temp[i]=nums[k-1-i];
        }
        for(int i=k;i<nums.length;i++){
            for(int j=0;j<k;j++){
                if(temp[j]<nums[i]){
                    change(temp,nums[i],j,k);
                    break;
                }
            }

        }
        return temp[k-1];
    }
    public static int[] change(int[] temp,int num,int j, int k){
        for(int i=k-1;i>j;i--){
            temp[i]=temp[i-1];
        }
        temp[j]=num;
        return temp;    
    }
}

后来,看了一种快排思想的思路。

这道题最好的解法应该是下面这种做法,用到了快速排序Quick Sort的思想,这里排序的方向是从大往小排。

核心思想是每次都要先找一个中枢点Pivot,然后遍历其他所有的数字,像这道题从大往小排的话,就把大于中枢点的数字放到左半边,把小于中枢点的放在右半边,这样中枢点是整个数组中第几大的数字就确定了,虽然左右两部分各自不一定是完全有序的,但是并不影响本题要求的结果,因为左半部分的所有值都大于右半部分的任意值,所以我们求出中枢点的位置。

如果中枢点的位置正好是k-1,那么直接返回该位置上的数字;如果大于k-1,说明要求的数字在左半部分,更新右边界,再求新的中枢点位置;反之则更新右半部分,求中枢点的位置;不得不说,这个思路真的是巧妙啊。。。

class Solution {
    public int findKthLargest(int[] nums, int k) {
        int left=0,right=nums.length-1;
        while(true){
            int pos=partition(nums,left,right);
            if(pos==k-1) return nums[pos];
            else if(pos>k-1) right=pos-1;
            else left=pos+1;
        }
    }
    public static int partition(int[] nums, int left,int right){
        int pivot=nums[left],l=left+1,r=right;
        int temp=0;
        while(l<=r){
            if(nums[l]<pivot && nums[r]>pivot){
                temp=nums[l];
                nums[l]=nums[r];
                nums[r]=temp;
                l++;
                r--;
            }
            if (nums[l] >= pivot) ++l;
            if (nums[r] <= pivot) --r;
        }        
        temp=nums[left];
        nums[left]=nums[r];
        nums[r]=temp;
        return r;
    }
}

其实时间上前一种12ms,后一种42ms。可能和pivot的选择关系比较大。

猜你喜欢

转载自blog.csdn.net/susuxuezhang/article/details/89284808
0条评论
添加一条新回复