算法通关村-----快速排序的应用

数组中的第K个最大元素

问题描述

给定整数数组 nums 和整数 k，请返回数组中第 k 个最大的元素。请注意，你需要找的是数组排序后的第 k 个最大的元素，而不是第 k 个不同的元素。详见leetcode215

问题分析

之前我们已经使用堆排序/堆查找的方式解决了这个问题。详见堆在查找和排序中的应用找到数组中的第K个最大元素，最直接的方式就是使用快速排序使数组有序，之后返回数组的第K个元素，当然可以使用快速排序来实现了，其实只要是需要数组有序，都可以通过快排来实现，因为快速排序相对高效。比如找最值，二分查找也需要数组有序等。

代码实现

可以使用快速排序的原理和实现中任一方式实现快速排序

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

public void quickSort(int[] nums,int left,int right){
    if(left>=right){
        return;
    }
    int pivot = nums[right];
    int i = left-1;
    for(int j=left;j<right;j++){
        if(nums[j]>pivot){
            i++;
            int temp = nums[j];
            nums[j] = nums[i];
            nums[i] = temp;
        }
    }
    int pivotIndex = i+1;
    int temp = nums[right];
    nums[right] = nums[pivotIndex];
    nums[pivotIndex] = temp;
    quickSort(nums,left,pivotIndex-1);
    quickSort(nums,pivotIndex+1,right);
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26

注意：题目中要求设计时间复杂度为O(n)的算法，所以有一个测试样例不通过。

快速排序的时间复杂度详见快速排序的原理和实现