1. 程式人生 > >python寫快排 寫了死迴圈, 記下來警醒一下

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 又永遠無法滿足。。。就是死迴圈了。