fix bug

Jack-Lee-Hiter · web-flow · commit 0a53804da466 · 2016-08-24T09:59:58.000+08:00
diff --git a/Target Offer/&#26368;&#23567;&#30340;k&#20010;&#25968;.py b/Target Offer/&#26368;&#23567;&#30340;k&#20010;&#25968;.py
@@ -57,14 +57,21 @@ def GetLeastNumbers(self, tinput, k):
             if len(output) < k:
                 output.append(number)
             else:
-                output = heapq.nsmallest(k, output)
-                if number >= output[-1]:
+                # &#26500;&#36896;&#26368;&#23567;&#22534;&#65292; &#19981;&#25512;&#33616;
+                # output = heapq.nsmallest(k, output)
+                # if number >= output[-1]:
+                #     continue
+                # else:
+                #     output[-1] = number
+                # &#26500;&#36896;&#26368;&#22823;&#22534;&#65292; &#25512;&#33616;
+                output = heapq.nlargest(k, output)
+                if number >= output[0]:
                     continue
                 else:
-                    output[-1] = number
-        return output
+                    output[0] = number
+        return output[::-1]     # &#26368;&#23567;&#22534;&#29992; return output
 tinput = [4,5,1,6,2,7,3,8]
 s = Solution()
 print(s.GetLeastNumbers_Solution(tinput, 4))
 print(s.GetLeastNumbers(tinput, 4))
-print(s.GetLeastNumbers(tinput, 5))
+print(s.GetLeastNumbers(tinput, 5))
