leetcode-快速排序

快速排序

通过一趟排序将待排序记录划分(partition)成独立的两部分，其中一部分记录的关键字均比另一部分记录的关键字小，则可以分别对着两部分记录继续进行排序，以达到整个序列有序。

代码实现

def quicksort(nums,left,right):
    if left>=right: return 

    def partition(nums,left,right):
        pivot = nums[left]
        i,j=left,right 

        while i<j:
            while i<j and nums[j]>=pivot:
                j-=1

            while i<j and nums[i]<=pivot:
                i+=1
            
            nums[i],nums[j] = nums[j],nums[i]

        nums[i],nums[left] = nums[left],nums[i]

        return i

    p = partition(nums,left,right)
    quicksort(nums,left,p-1)
    quicksort(nums,p+1,right)

快排-topk问题

• 原理见leetcode-topk

1. 第k小的数

def topk(nums,k):
    def quicksort(nums,left,right):

        pivot = nums[left]
        i,j=left,right 

        while i<j:
            while i<j and nums[j]>=pivot:
                j-=1
            while i<j and nums[i]<=pivot:
                i+=1
            nums[i],nums[j] = nums[j],nums[i]
			
        nums[i],nums[left] = nums[left],nums[i]

        if k<i:
            return quicksort(nums,left,i-1)
        if k>i:
            return quicksort(nums,i+1,right)

        return nums[k-1]

    return quicksort(nums,0,len(nums)-1)

2. 第k大的数

class Solution:
    def findKth(self , a: List[int], n: int, K: int) -> int:
        # write code here

        def quicksort(a,left,right):
            pivot = a[left]
            i,j = left,right 

            while i<j:
                while i<j and a[j]>=pivot:
                    j-=1
                while i<j and a[i]<=pivot:
                    i+=1
                a[i],a[j]=a[j],a[i]
            
            a[i],a[left]= a[left],a[i]

            if i==n-K:
                return a[i]

            elif i<n-K:
                return quicksort(a,i+1,right)
            else:
                return quicksort(a,left,i-1)

        return quicksort(a,0,n-1)