Commit0a53804

authored

fix bug

1 parent09230bf commit0a53804Copy full SHA for 0a53804

File tree

+12

-5

lines changed

+12

-5

lines changed

Lines changed: 12 additions & 5 deletions

Original file line number	Diff line number	Diff line change
`@@ -57,14 +57,21 @@ def GetLeastNumbers(self, tinput, k):`
`57`	`57`	`iflen(output)<k:`
`58`	`58`	`output.append(number)`
`59`	`59`	`else:`
`60`		`-output=heapq.nsmallest(k,output)`
`61`		`-ifnumber>=output[-1]:`
	`60`	`+# 构造最小堆，不推荐`
	`61`	`+# output = heapq.nsmallest(k, output)`
	`62`	`+# if number >= output[-1]:`
	`63`	`+# continue`
	`64`	`+# else:`
	`65`	`+# output[-1] = number`
	`66`	`+# 构造最大堆，推荐`
	`67`	`+output=heapq.nlargest(k,output)`
	`68`	`+ifnumber>=output[0]:`
`62`	`69`	`continue`
`63`	`70`	`else:`
`64`		`-output[-1]=number`
`65`		`-returnoutput`
	`71`	`+output[0]=number`
	`72`	`+returnoutput[::-1]# 最小堆用 return output`
`66`	`73`	`tinput= [4,5,1,6,2,7,3,8]`
`67`	`74`	`s=Solution()`
`68`	`75`	`print(s.GetLeastNumbers_Solution(tinput,4))`
`69`	`76`	`print(s.GetLeastNumbers(tinput,4))`
`70`		`-print(s.GetLeastNumbers(tinput,5))`
	`77`	`+print(s.GetLeastNumbers(tinput,5))`

Comments

(0)