leetcode刷题--215. 数组中的第K个最大元素

题目描述

在未排序的数组中找到第 k 个最大的元素。请注意，你需要找的是数组排序后的第 k 个最大的元素，而不是第 k 个不同的元素。

输入: [3,2,1,5,6,4] 和 k=2输出: 5

 求解思路1

刚开始想到的是用冒泡排序，循环k次，数组中第len-k个就是第k个最大元素。但复杂度比较高，内存利用率也比较大。

代码1

class Solution {
    public int findKthLargest(int[] nums, int k) {
        int len = nums.length;
        for(int i=0;i<k;i++){  //循环k次
            for(int j=0;j<len-i-1;j++){
                if(nums[j]>nums[j+1]){
                    int swap = nums[j];
                    nums[j] = nums[j+1];
                    nums[j+1] = swap;
                }
            }
        }
        return nums[len-k];
    }
}

求解思路2

使用快速排序的思想：（本来以为会快一点，结果性能并没有提高多少）

1. 第一个作为基准数
2. 大的放在左边，小的放在右边
3. 判断最后的p
4. 若=k-1，则输出nums[p]
5. >k-1，在左边继续找
6. <k-1，在右边继续找

代码

class Solution {
    public int re = 0;
    public int findKthLargest(int[] nums, int k) {
        int len = nums.length;
        quicksort(nums,k,0,len-1);
        return re;
    }
    void quicksort(int[] nums,int k,int left,int right){
        if(left<=right){
            int lo = partiton(nums,left,right);  //判断一次快排后的p的位置
            if(lo == k-1){
                re = nums[lo];    //等于直接输出
            }
            else if(lo<k-1){
                quicksort(nums,k,lo+1,right);  //小于在右边找
            }
            else if(lo>k-1){
                quicksort(nums,k,left,lo-1);    //大于在左边找
            }
        }
    }
    int partiton(int[] nums,int left,int right){ //快排
        int flag = nums[left];
        int i=left+1;
        int p=right;
        while(i<=p){
            while(nums[i]>flag&&i<right){   //两个指针
                i++;
            }
            while(nums[p]<flag){
                p--;
            }
            if(i<p){
                int swap = nums[i];
                nums[i] = nums[p];
                nums[p] = swap;
                i++;
                p--;
            }
            else{
                break;
            }
        }
        nums[left] = nums[p];
        nums[p] = flag;
        return p;
    }
}