程序员

leetcode 249--map用法

2018-12-15  本文已影响4人  Ariana不会哭

这道题使用C++ java 分别写出来,比较运行速度。


图片.png

C++:
算法是很简单的,当前位减第一位:a[0]。如果出现负数怎么办?(cc - a[0] + 26) % 26) 这样"za"就等于25 而不是-1.
当然这道题默认 “za”="ab" 可能以后会有followup 避免这个情况。
t = t + to_string((cc - a[0] + 26) % 26) + ",";
注意有逗号: 引入都好的原因:“2,5” 和“25” 作区别

举个例子:
“acef” --索引key是: “2,4,5,”
“abc” --索引key是: “1,2,”

//my
vector<vector<string>> groupStrings(vector<string>& strings) {
    vector<vector<string>> ans;
    unordered_map<string, multiset<string>>m;
    for (auto a : strings) {
        string t = "";
        for (char cc : a) {
            t = t + to_string((cc - a[0] + 26) % 26) + ",";
        }
        m[t].insert(a);
    }

    for (auto it = m.begin(); it != m.end(); it++) {
        ans.push_back(vector<string>(it->second.begin(), it->second.end()));
    }
    return ans;
}

java:
思路与过程和C++是一致的,主要使用了stringbuilder,这个里面append方法速度很快。
但是使用哈希表的时候没有C++简单,他没有索引,我们需要手动添加key.还要判断是否已经有索引。
第三,最后建立完成哈希地图后,需要把结果添加到res中。

//faster:
    public List<List<String>> groupStrings2(String[] strings) {
        List<List<String>> res= new ArrayList<List<String>>();
        HashMap<String, ArrayList<String>> m = new HashMap<String, ArrayList<String>>();

        for (String a : strings) {
            StringBuilder sb = new StringBuilder();
            for (char c : a.toCharArray()) {
                sb.append((c + 26 - a.charAt(0)) % 26);
                sb.append(",");
            }
            String t =sb.toString();
            if(m.containsKey(t))
                m.get(t).add(a);
            else {
                m.put(t, new ArrayList<>());
                m.get(t).add(a);
            }
        }
        res.addAll(m.values());
        
        return res;
    }
上一篇下一篇

猜你喜欢

热点阅读