LeetCode 面试题 10.02. 变位词组[map] HERODING的LeetCode之路

Posted 2021-08-15 HERODING23

解题思路：
看似比较复杂的题目，其实换一个思路就很简单，遍历所有的字符串，然后sort，这样只要是变位词，sort 后都相同，统计他们的下标，最后根据他们的下标放入同一个vector中，代码如下：

class Solution {
public:
    vector<vector<string>> groupAnagrams(vector<string>& strs) {
        unordered_map<string, vector<int>> mp;
        // 排序 + 存储下标
        for(int i = 0; i < strs.size(); i ++) {
            string s = strs[i];
            sort(s.begin(), s.end());
            mp[s].push_back(i);
        }
        vector<vector<string>> res;
        // 统计
        for(auto& [a, b] : mp) {
            vector<string> ans;
            for(int& i : b) {
                ans.push_back(strs[i]);
            }
            res.push_back(ans);
        }
        return res;
    }
};