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

写 寻找最小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 又永远无法满足。。。就是死循环了。

猜你喜欢

转载自blog.csdn.net/tiny_fisher/article/details/85077590
今日推荐