Update Leetcode 题解.md

CyC2018 · CyC2018 · commit cc9dd446f2e4 · 2019-02-26T15:14:52.000+08:00
diff --git a/docs/notes/Leetcode 题解.md b/docs/notes/Leetcode 题解.md
@@ -403,10 +403,12 @@ public List<Integer> topKFrequent(int[] nums, int k) {
     }
     List<Integer> topK = new ArrayList<>();
     for (int i = buckets.length - 1; i >= 0 && topK.size() < k; i--) {
-        if (buckets[i].size() <= (k - topK.size())) {
-            topK.addAll(buckets[i]);
-        } else {
-            topK.addAll(buckets[i].subList(0, k - topK.size()));
+        if (buckets[i] != null) {
+            if (buckets[i].size() <= (k - topK.size())) {
+                topK.addAll(buckets[i]);
+            } else {
+                topK.addAll(buckets[i].subList(0, k - topK.size()));
+            }
         }
     }
     return topK;

Original file line number	Diff line number	Diff line change
`@@ -403,10 +403,12 @@ public List<Integer> topKFrequent(int[] nums, int k) {`
`403`	`403`	`}`
`404`	`404`	`List<Integer> topK = new ArrayList<>();`
`405`	`405`	`for (int i = buckets.length - 1; i >= 0 && topK.size() < k; i--) {`
`406`		`- if (buckets[i].size() <= (k - topK.size())) {`
`407`		`- topK.addAll(buckets[i]);`
`408`		`- } else {`
`409`		`- topK.addAll(buckets[i].subList(0, k - topK.size()));`
	`406`	`+ if (buckets[i] != null) {`
	`407`	`+ if (buckets[i].size() <= (k - topK.size())) {`
	`408`	`+ topK.addAll(buckets[i]);`
	`409`	`+ } else {`
	`410`	`+ topK.addAll(buckets[i].subList(0, k - topK.size()));`
	`411`	`+ }`
`410`	`412`	`}`
`411`	`413`	`}`
`412`	`414`	`return topK;`