LeetCode0215.数组中的第K个最大元素 Go语言AC笔记

 时间复杂度：O(n)

解题思路

该题为快速排序的变种题，用的是快速选择算法。

主要思路就是参考快速排序，每一趟找到第x个大的元素，从大到小排序。比较x与k的大小，如果x小于k就说明我们要找的第k个元素在当前已知第x大元素的右边，我们只需对x右侧快排即可；同理如果x比k大，就说明第k大的元素在第x大的元素的左边，我们只需对x左侧快排即可。

不过该题需要注意的是由于题目要求时间复杂度为O(n)，所以我们在partition时注意要随机取一个元素当作标准，而不能只选择第一个元素当作标准数。根据算法导论的证明这样时间复杂度能降为O(n)，否则时间复杂度可能为O(n²)。

另外该题快速选择函数务必为递归写法，我一开始用的是循环每次都是超时，后来改成递归就顺利AC了。

AC代码

func findKthLargest(nums []int, k int) int {
    return quickSelect(nums,0,len(nums)-1,k)
}

func quickSelect(nums []int,left,right,k int)int{
    index:=partition(nums,left,right)
    if k-1==index{
        return nums[index]
    }
    if k-1<index{
        return quickSelect(nums,left,index-1,k);
    }else{
        return quickSelect(nums,index+1,right,k);
    }
}

func partition(nums []int,left,right int)int{
    k:=left+rand.Intn(right-left+1)
    nums[left],nums[k]=nums[k],nums[left]
    base:=nums[left]
    for left<right{
        for left<right&&nums[right]<=base{
            right--
        }
        nums[left]=nums[right]
        for left<right&&nums[left]>=base{
            left++
        }
        nums[right]=nums[left]
    }
    nums[left]=base
    return left
}

感悟

该题是我第一次面试时遇到的题目，当时也没刷力扣题，傻呵呵地没想出来解题方案，面试官提醒我用快速排序，给我很细致地讲了思路，让我自己实现一下。之后我根据面试官给的思路很顺利的写完了代码，虽然面试挂了，但是很感谢那位面试官，让我对该题目的印象格外深刻。

partiton函数中租后一定要把base放回nums数组中