简单粗暴——快速排序介绍(Python实现LeetCode215)

快速排序

快速排序(Quick Sort)算法也是一种递归排序,对冒泡排序的改进,分而治之,减少比较次数,牺牲空间。

原理

设置一个基准值(pivot),这个基准点可以随意设置,但是一般来说我们选择第一个数据。
将列表分成左右两部分,一部分比基准大一部分比基准小。
采用挖坑法:
1、用left标识数组的第一个数据,right标识数组的最后一个数据
2、如果right位上的数据大于基准值,则继续往后遍历(right- -);’如果小于基准值,则将right位上的值赋值给left位。
3、让left往前遍历, 如果left位上的数据小于基准值,则继续往前遍历(left++);如果left位上的值大于基准值,则将left位上的值赋值给right位。
4、直到left==right这一趟排序就结束了,将基准值赋值给left位
5、再采用递归的方法将基准值位左边与右边的两部分数组分别进行快速排序。前半部分和后半部分均有序时该数组就自然有序了。
在这里插入图片描述

例题

LeetCode 215题:求数组中第K大元素
示例 1:

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

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

代码

class Solution:
    def findKthLargest(self, nums, k):
        self._k = len(nums) - k  # 第K大
        return self.quicksort(nums, 0, len(nums) - 1)

    def quicksort(self, nums, left, right):
        if left == right:
            return nums[left]
        pivot = self.partiton(nums, left, right)  # pivot位置
        # print("p", pivot)
        if pivot == self._k:
            return nums[pivot]  # 判断是否为第K大,是的话返回值
        elif pivot < self._k:
            return self.quicksort(nums, pivot + 1, right)
        else:
            return self.quicksort(nums, left, pivot - 1)

    def partiton(self, nums, left, right):  # 确认基准元素最终位置
        pivot = nums[left]
        i, j = left, right
        while i < j:  # 判断元素与基准元素的大小关系,挖坑填坑
            while i < j and nums[j] >= pivot:
                j -= 1
            if i < j:
                nums[i] = nums[j]
                i += 1
            while i < j and nums[i] <= pivot:
                i += 1
            if i < j:
                nums[j] = nums[i]
                j -= 1
        nums[i] = pivot
        # print(i, pivot)
        return i


if __name__ == '__main__':
    s = Solution()
    s = s.findKthLargest([3, 2, 1, 5, 6, 4], 2)
    print(s)

简便实现

class Solution: 
    def findKthLargest(self, nums: List[int], k: int) -> int: 
        nums.sort() 
        return nums[len(nums) - k] 
©️2020 CSDN 皮肤主题: 书香水墨 设计师:CSDN官方博客 返回首页