Create Solution.java

Author: Jerring

solution/0347.Top K Frequent Elements/Solution.java

@@ -0,0 +1,24 @@
+class Solution {
+    public List<Integer> topKFrequent(int[] nums, int k) {
+        Map<Integer, Integer> map = new HashMap<>();
+        for (int num : nums) {
+            map.put(num, map.getOrDefault(num, 0) + 1);
+        }
+        List<Integer>[] buckets = new ArrayList[nums.length + 1];
+        for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
+            int num = entry.getKey();
+            int count = entry.getValue();
+            if (buckets[count] == null) {
+                buckets[count] = new ArrayList<>();
+            }
+            buckets[count].add(num);
+        }
+        List<Integer> topK = new ArrayList<>(k);
+        for (int i = buckets.length - 1; i >= 0 && topK.size() < k; --i) {
+            if (buckets[i] != null) {
+                topK.addAll(buckets[i]);
+            }
+        }
+        return topK;
+    }
+}