edit leetcode 347

This commit is contained in:
xPluto
2019-02-26 02:37:49 +08:00
parent 55b8121d93
commit 9b22f5e6af

View File

@ -404,7 +404,11 @@ 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] != null) {
if (buckets[i].size() <= (k - topK.size())) {
topK.addAll(buckets[i]);
} else {
topK.addAll(buckets[i].subList(0, k - topK.size()));
}
}
}
return topK;