当前位置: 代码迷 >> 综合 >> 给定一个非空的整数数组,返回其中出现频率前 k 高的元素。要求时间复杂度优于O(nlogn)
  详细解决方案

给定一个非空的整数数组,返回其中出现频率前 k 高的元素。要求时间复杂度优于O(nlogn)

热度:60   发布时间:2023-10-15 04:55:05.0

给定一个非空的整数数组,返回其中出现频率前 k 高的元素。要求时间复杂度优于O(nlogn)
示例 1:
输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]
示例 2:
输入: nums = [1], k = 1
输出: [1]
说明:
你可以假设给定的 k 总是合理的,且 1 ≤ k ≤ 数组中不相同的元素的个数。

桶排序

public List<Integer> topKFrequent(int[] nums, int k) {
    HashMap<Integer, Integer> map = new HashMap<>();for(int i : nums) {
     // 统计频率,key为元素,value为次数map.put(i, map.getOrDefault(i, 0) + 1);}// 创建nums.length + 1个桶List<Integer>[] bucket = new List[nums.length + 1];// 遍历map,根据value值放到对应的桶中for(Map.E***y<Integer, Integer> e : map.e***ySet()) {
    Integer value = e.getValue();if(bucket[value] == null) {
    bucket[value] = new ArrayList<Integer>();}bucket[value].add(e.getKey());}List<Integer> freList = new ArrayList<>();// 桶的编号表示出现次数,所以倒数桶for(int j = bucket.length - 1; j > -1 && freList.size() < k; j--) {
    if(bucket[j] != null)freList.addAll(bucket[j]);}return freList;}

==================================

下面自己练习一下集合

 public static void main(String[] args) {
    int[] arr = {
    1,1,1,1,1,1,2,2,2,2,2,3};int k = 2;getResult(arr,k);}private static void getResult(int[] arr, int k) {
    HashMap<Integer,Integer> map = new HashMap<>();List<Integer>[] listArr = new List[arr.length+1];for (int i=0;i<arr.length;i++){
    map.put(arr[i],map.getOrDefault(arr[i],0)+1);}for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
    if (listArr[entry.getValue()]==null){
    listArr[entry.getValue()]=new ArrayList<>();}listArr[entry.getValue()].add(entry.getKey());}List<Integer> ls = new ArrayList<>();for (int i=listArr.length-1;i>=0&&ls.size()<k;i--){
    if (listArr[i]!=null){
    ls.addAll(listArr[i]);}}System.out.println(ls);}

=====================================

 public static void main(String[] args) {
    int[] arr = {
    1,1,1,2,2,2,2,2,3};
// HashMap<Integer,Integer> hashMap = new HashMap<>();
// hashMap.put(1,123);
// System.out.println(hashMap.get(1));getResult(arr);}private static void getResult(int[] arr) {
    HashMap<Integer,Integer> map = new HashMap<>();HashSet<Integer> set = new HashSet<>();for (int i=0;i<arr.length;i++){
    set.add(arr[i]);}Iterator<Integer> iterator = set.iterator();while (iterator.hasNext()){
    Integer temp = iterator.next();for (int j=0;j<arr.length;j++){
    if (temp==arr[j]){
    if (map.get(temp)==null){
    map.put(temp,1);}else {
    
// Integer lastValue = map.get(temp);
// map.put(temp,++lastValue);map.put(temp,map.get(temp)+1);}}}}System.out.println(map.toString());Set<Map.Entry<Integer, Integer>> entrySet = map.entrySet();ArrayList<Map.Entry<Integer, Integer>> list = new ArrayList<>(entrySet);System.out.println(list);Collections.sort(list, new Comparator<Map.Entry<Integer, Integer>>() {
    @Overridepublic int compare(Map.Entry<Integer, Integer> o1, Map.Entry<Integer, Integer> o2) {
    return o2.getValue() - o1.getValue();}});System.out.println(list);List<Integer> ls = new ArrayList<>();for (int i=0;i<2;i++){
    ls.add(list.get(i).getKey());}System.out.println(ls);}
  相关解决方案