【Leetcode】Top K Frequent Elements

题目链接:https://leetcode.com/problems/top-k-frequent-elements/

Given a non-empty array of integers, return the k most frequent elements.

For example,
Given [1,1,1,2,2,3] and k = 2, return [1,2].

Note:

You may assume k is always valid, 1 ≤ k ≤ number of unique elements.
Your algorithm’s time complexity must be better than O(n log n), where n is the array’s size.

思路:

  1. 定义一个类 保存一个num的值和频率,并实现比较方法

  2. 遍历数组,并用hashMap保存各个数字出现的频率

  3. 对hashmap的value集合排序,获取频率最高的k位

算法:

    public List topKFrequent(int[] nums, int k) {
        List res = new ArrayList();
        Map map = new HashMap();
        //统计各数字出现的次数 存入hashmap
        for(int t:nums){
            FEntity f;
            if(map.containsKey(t)){
                f = map.get(t);
                f.frequent++;
            }else{
                f = new FEntity(t, 1);
            }
            map.put(t, f);
        }
        //对hashmap的value 按照频率进行排序
        List values = new ArrayList();
        Set keys = map.keySet();
        for(Integer key:keys){
            values.add(map.get(key));
        }
        Collections.sort(values);
        //统计频率最高的k个数
        for(int i=0;i1-i).num);
        }
        return res;
    }

    class FEntity implements Comparable {
        int num, frequent;

        public FEntity(int num, int frequent) {
            this.num = num;
            this.frequent = frequent;
        }

        public int compareTo(FEntity arg0) {
            if (frequent > arg0.frequent)
                return 1;
            else if (frequent == arg0.frequent)
                return 0;
            else
                return -1;
        }
    }

你可能感兴趣的:(leetcode,Leetcode题解java版)