LeetCode-49. 字母异位词分组

49. 字母异位词分组

给定一个字符串数组,将字母异位词组合在一起。字母异位词指字母相同,但排列不同的字符串。

示例:

输入: ["eat", "tea", "tan", "ate", "nat", "bat"],
输出:
[
  ["ate","eat","tea"],
  ["nat","tan"],
  ["bat"]
]

说明:

  • 所有输入均为小写字母。
  • 不考虑答案输出的顺序。

哈希表

虽然过了,但不足的地方在于最后vec空间的处理,因为vector不预先分配空间,无法使用下标法的,所以先分足够的空间,再最后重新分配空间。

static const int _ = []() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    return 0;
}();
class Solution {
public:
    vector<vector<string>> groupAnagrams(vector<string>& strs) {
        unordered_map<string,int> mp;
        vector<vector<string>> vec(strs.size());
        int k=0;
        for(int i=0;i<strs.size();i++)
        {
            string ss = strs[i];
            sort(ss.begin(),ss.end());
            if(mp.count(ss)==0)
            {
                mp[ss]=k;
                k++;
            }
            vec[mp[ss]].push_back(strs[i]);
        }
        vec.resize(k);
        return vec;
    }
};

改良

这里对哈希表做了改良,改良后的哈希表表现形式应该是这样的:

mp["abc"] = {"abc","bac","cba"};
mp["tea"] = {"tea","tae","ate"};
...
static const int _ = []() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    return 0;
}();
class Solution {
public:
    vector<vector<string>> groupAnagrams(vector<string>& strs) {
        unordered_map<string,vector<string>> mp;
        vector<vector<string>> vec;
        for(auto r:strs)
        {
            string s(r);
            sort(s.begin(),s.end());
            mp[s].push_back(r);
        }
        for(auto p:mp)
        {
            vec.push_back(p.second);
        }
        return vec;
    }
};

猜你喜欢

转载自blog.csdn.net/qq_41822647/article/details/88111820