python寫快排 寫了死迴圈, 記下來警醒一下
阿新 • • 發佈:2019-01-02
寫 尋找最小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 又永遠無法滿足。。。就是死迴圈了。