python写快排写了死循环，记下来警醒一下

最新推荐文章于 2024-06-14 21:05:53 发布

原创最新推荐文章于 2024-06-14 21:05:53 发布 · 988 阅读

0 ·

CC 4.0 BY-SA版权

python 专栏收录该内容

5 篇文章

订阅专栏

在编写寻找最小k个数的代码时，使用快速排序算法出现错误。问题出在未添加两个等号，导致左指针和右指针在相等时无法移动，造成死循环。正确做法应确保在满足left > right条件时，左右指针能继续移动以完成排序。

写寻找最小k 个数时，直接用快排写。但是快排一开始写错了。。。

class Solution:
    
    def GetLeastNumbers_Solution(self, tinput, k):
        if tinput ==[] or k > len(tinput) or k<0:
            return []
        #tinput.sort()
        self.quicksort(tinput,0,len(tinput)-1)
        
        return tinput[0:k]
        
    def quicksort(self,alist,start,end):
        if start <end:
            pivot = self.partition(alist, start,end)
            self.quicksort(alist,start,pivot-1)
            self.quicksort(alist,pivot+1,end)
        
        
    def partition(self,alist,start,end ):
        
        pivotvalue = alist[start]
        left = start+1;
        right = end
        
        done = False
        while not done:
            
            # 出错点： 两个 = 没写
            
            #while left < right and alist[left]<=pivotvalue:
            while left <= right and alist[left]<=pivotvalue:
                left +=1;
                
                
            #while left <= right and alist[right]>= pivotvalue:    
            while left <= right and alist[right]>= pivotvalue:
                right -=1
            
            if left>right:
                done = True
            else:
                alist[left],alist[right] = alist[right],alist[left]
                
        
        alist[start],alist[right] =  alist[right],alist[start] 
        
        return right
    
myt = Solution()
print(myt.GetLeastNumbers_Solution([12,43,3,5,2],2))

出错点是两个= 没加，会造成什么影响呢？

首先要明确的是，最后要让左指针直到排好之后的第一个大于分位数的数，右指针指向最后一个小于分位数的数；也就是判断语句 left > right:

那么，前面操作两个指针的时候，就要在相同的时候还能移动呀，不然，相同了，左右指针都不动了，然后left>right 又永远无法满足。。。就是死循环了。