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