力扣【347】前 K 个高频元素

题目:

给定一个非空的整数数组,返回其中出现频率前 k 高的元素。

示例 1:

输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]
示例 2:

输入: nums = [1], k = 1
输出: [1]

题解:使用HashMap

小知识:快速对齐快捷键是Ctrl+Alt+L

import java.util.*;

//快速对齐快捷键是Ctrl+Alt+L
class Solution {
    public int[] topKFrequent(int[] nums, int k) {
        int[] res = new int[k];    // 结果数组
        Map<Integer, Integer> map = new HashMap();
        // 统计数组中各元素出现的次数
        for (int num : nums) {
            if (map.containsKey(num)) {
                map.put(num, map.get(num) + 1);
            } else {
                map.put(num, 1);
            }
        }

        //找出现元素最多的元素,注意Map.Entry(Integer, Integer)的Map首字母大写,注意加(),注意是<>
        int maxTimes = 0;
        for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
            if (entry.getValue() > maxTimes) {
                maxTimes = entry.getValue();
            }
        }
        //从大到小放到结果数组里面,注意entry.getKey()和entry.getValue()
        int i = 0;
        while (i < k) {
            for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
                if (entry.getValue() == maxTimes) {
                    res[i] = entry.getKey();
                    i++;//这里不加break是因为有可能有重复的最大次数的值。
                }
            }
            maxTimes--;
        }
        return res;
    }
}

public class Main {
    public static void main(String[] args) {
        int[] nums = {1, 1, 1, 2, 2, 3};
        int k = 2;
        Solution p = new Solution();
        int[] res = p.topKFrequent(nums, k);
        System.out.println("输出结果" + Arrays.toString(res));
    }
}

猜你喜欢

转载自blog.csdn.net/qq1922631820/article/details/113976843