[LeetCode] 692.Top K Frequent Words之TreeMap与HashMap

[LeetCode] 692.Top K Frequent Words之TreeMap与HashMap

[LeetCode] 692.Top K Frequent Words

题意

输入： 一个字符串数组、一个整数值（K）
输出： 频率最高的前K个字符串的List集合

思路

1）遍历一遍数组，用Map（标记为hashmap1）统计每个字符串出现的次数，以字符串为key，次数为value。
2）设置一个K大小的最大堆，一个以次数为key，List为value的Map（标记为hashmap2），List里面存放该次数的所有字符串。
3）遍历hashmap1的所有key值，若hashmap2不含有值为hashmap1.get(key)的key值，则将key加入最大堆，并new出List，将key加入List中。
4）最后，遍历最大堆，取出前K个字符串，返回结果。

代码

public List<String> topKFrequent(String[] words, int k) {
        List<String> res=new ArrayList<>();
        Map<String,Integer> m=new HashMap<>();//出错点
        for(int i=0;i<words.length;i++){
            if(!m.containsKey(words[i]))
                m.put(words[i],1);
            else
                m.put(words[i],1+m.get(words[i]));
        }
        PriorityQueue<Integer> pq=new PriorityQueue<>(k,Comparator.reverseOrder());
        Map<Integer,List<String>> map=new HashMap<>();
        Iterator it = m.entrySet().iterator();
        while (it.hasNext()) {
            Map.Entry entry = (Map.Entry) it.next();
            String key =(String) entry.getKey();
            int value =(int) entry.getValue();
            if (!map.containsKey(value)){
                pq.offer(value);
                List<String> ls=new ArrayList<>();
                ls.add(key);
                map.put(value,ls);
            }else
                map.get(value).add(key);
        }
        while(!pq.isEmpty()&& res.size()<k){
            int key=pq.poll();
            for(int i=0;i<map.get(key).size() && res.size()<k;i++)
                res.add(map.get(key).get(i));
        }
        return res;
    }

结果

思路其实是正确的，代码也差不多正确，但是有相同频次的字符串无法保证相对位置顺序的错误，这是由于第一个Map采用了HashMap的原因，换成TreeMap就可以保证相对位置顺序了。

修改

Map<String,Integer> m=new HashMap<>();//出错点
//换成
TreeMap<String,Integer> m=new TreeMap<>();//保证次数相同的字符串相对位置顺序

HashMap与TreeMap

HashMap

HashMap是基于哈希表的Map接口的实现。此实现提供所有可选的映射操作，并允许使用null值和null键。除了非同步和允许使用null之外，HashMap类和Hashtable大致相同。HashMap不保证映射的顺序，特别是它不保证该顺序恒久不变。

HashMap 的实例有两个参数影响其性能：初始容量 和加载因子。

TreeMap

TreeMap是基于红黑树来实现的，没有调优选项，因为该树总处于平衡状态。

1) TreeMap(): 构建一个空的Map树
2) TreeMap(Map m): 构建一个Map树，并添加Map m中的所有元素
3) TreeMap(Comparator c): 构建一个Map树，并且使用特定的比较器对关键字进行排序
4) TreeMap(SortedMap s): 构建一个Map树，添加Map树s中所有映射，并使用与有序映像s相同的比较器排序

TreeMap映射一般根据其键的自然顺序进行排序（可以保证相对位置），或者根据创建映射时提供的Comparator进行排序，具体取决于使用的构造方法。

HashMap和TreeMap比较

1）HashMap适用于在Map中插入、删除和定位元素。
2）TreeMap适用于按自然顺序或者自定义顺序遍历键（key）。
3）HashMap通常比TreeMap快一点（树和哈希表的数据结构使然），建议多使用HashMap，在需要排序的Map时候才使用TreeMap。
4）两者都非线程安全（？）。
5）HashMap的结果是没有排序的，而TreeMap输出的结果是排好序的。

更多HashMap、TreeMap的相关源码、线程安全性参考：传送门。