LeetCode215. 数组中的第K个最大元素

1、题目

给定整数数组 nums 和整数 k，请返回数组中第 k 个最大的元素。

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

示例 1:

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

2、解答

• 暴力法：直接调用数组排序算法，然后返回对应的元素即可。
• 快速排序法：这个就是自己写一下快排算法。快排分为单指针遍历和双指针遍历。但是思想都是选这一个元素作为基础节点，然后遍历数组把小于基础节点的元素放到左边，大于的放到右边。双指针遍历需要注意的点事，如果选取左边节点为基础节点，那么先从右边开始遍历，因为如果从左边开始遍历，没办法保证停止遍历时左指针指向的元素大于基础节点，因为还可能都小于这个基础节点。单指针就是找到每次基础节点应该插入的位置。

class Solution(object):
    def findKthLargest(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: int
        """
        self.quick_sort2(nums, 0, len(nums) - 1)
        return nums[-k]

    def quick_sort(self, nums, start, end):
        """
        快速排序双指针法：需要注意的是如果选择左边第一个节点为基础值得话，那么要右边的指针先移动，
        否则会出现问题，因为左边先动无法保证跳出循环时左指针所指向的值就一点大于基础值
        :param nums:
        :param start:
        :param end:
        :return:
        """
        if start >= end:
            return
        left = start
        right = end
        base = nums[left]
        while start < end:
            while start < end and nums[end] > base:
                end -= 1
            while start < end and nums[start] <= base:
                start += 1

            if start < end:
                nums[start], nums[end] = nums[end], nums[start]
        nums[left], nums[start] = nums[start], nums[left]
        self.quick_sort(nums, left, start - 1)
        self.quick_sort(nums, start + 1, right)

    def quick_sort2(self, nums, start, end):
        """
        快速排序单指针法：选择右边节点为基础节点，遍历整个数组，把小于基础节点的都放到左边
        :param nums:
        :param start:
        :param end:
        :return:
        """
        if start >= end:
            return
        left = start - 1
        base = nums[end]
        if start < end:
            for i in range(start, end):
                if nums[i] < base:
                    left += 1
                    nums[left], nums[i] = nums[i], nums[left]
            nums[left + 1], nums[end] = nums[end], nums[left + 1]
            self.quick_sort2(nums, start, left)
            self.quick_sort2(nums, left + 2, end)