哈希表与字符串--03-字母异位词分组[中等]

力扣

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

示例:

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

输出:

[

["ate","eat","tea"],

["nat","tan"],

["bat"]

]

说明：

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

思路

给所有输入由小到大排序，作为key，value从0顺序增大生成。每次遇到相同的key，就可以找分组，将单词进行添加了。

答案

#include <iostream>
#include <map>
#include <string>
#include <vector>
#include <stack>
#include <set>
#include <algorithm> //标准算法的头文件
using namespace std;

class Solution {
public:
	vector<vector<string>> groupAnagrams(vector<string>& strs) {
		map<string, int> m;
		int cnt = 0; //计数value
		vector<vector<string>> ret;
		vector<string> v; //仅用于分配空间
		for (int i = 0; i < strs.size(); i++) {
			string temp = strs[i];
			sort(temp.begin(), temp.end()); //对每个单词排序
			if (m.find(temp) == m.end()) { //表示未添加
				m.insert(make_pair(temp, cnt)); //加入哈希
				ret.push_back(v); //添加空间
				ret[cnt].push_back(strs[i]); //添加结果
				cnt = cnt + 1; 
			}
			else {  //如果查找到
				ret[m[temp]].push_back(strs[i]); //添加结果
			}

		}
		return ret;
	}
};


void test02()
{
	vector<string> strs = { "eat", "tea", "tan", "ate", "nat", "bat" };
	Solution s;
	vector<vector<string>> ret = s.groupAnagrams(strs);
	for (int i = 0; i < ret.size(); i++) {
		for (int j = 0; j < ret[i].size(); j++)
			cout << ret[i][j] << "\t";
		cout << endl;
	}
}

int main()
{
	test02();

	system("pause");
	return 0;
}

class Solution {
public:
    vector<vector<string>> groupAnagrams(vector<string>& strs) {
        vector<vector<string>> ret;
        unordered_map<string,int> umap;
        int i=0;
        for(string item:strs) {
            string t=item;
            sort(t.begin(),t.end());
            if(umap.find(t)==umap.end()) {
                umap.insert(make_pair(t,i++));
                vector<string> v;
                v.push_back(item);
                ret.push_back(v);
            } else {
                ret[umap[t]].push_back(item);
            }
        }
        return ret;
    }
};