LeetCode.49. 字母异位词分组

题目

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

示例:

输入:

[“eat”, “tea”, “tan”, “ate”, “nat”, “bat”],

输出:

[
[“ate”,”eat”,”tea”],
[“nat”,”tan”],
[“bat”]
]

说明：

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

思路

首先准备

• 一个动态二维数组 ans，存放最终答案
• 一个无序图 m，存放排序后的字符串和其索引
• 一个动态一位数组 str，copy 传来的形参

把每次遇到的字符串都排个序，如果是字母异位词，肯定排序结果一样，如果 m 中不存在，就需要额外存排序后的该字符串和其索引，再把未排序的字符串添加进相应的动态数组后面。

代码

class Solution {
public:
    vector<vector<string>> groupAnagrams(vector<string>& strs) {
        vector<vector<string>> ans;
        unordered_map<string,int> m;
        int k=0;
        vector<string> str = strs;
        for(int i=0;i<strs.size();i++){
            sort(str[i].begin(),str[i].end());
            if(!m.count(str[i])){
                m[str[i]] = k++;
                ans.push_back(vector<string>{});
            }
            ans[m[str[i]]].push_back(strs[i]); // str[i] 就是有序数组， m[str[i]]就是其索引
        }
        return ans;
    }
};