C++算法:字母异位词分组-----multimap和map的使用

题目:

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

示例:

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

说明:

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

方法一:

使用multimap的特性,自动排序,key值可以重复!!!

代码:


class Solution {
public:
    vector<vector<string>> groupAnagrams(vector<string>& strs) {
        
        map<multimap<char, int>, int> a;
        
        multimap<char, int> temp;
        
        vector<vector<string>> ret;
        
        int i, j, count=0;
        for(i = 0; i < strs.size(); i++)
        {
            temp.clear();
            for(j = 0; j < strs[i].size(); j++)
                temp.insert(make_pair(strs[i][j],1));
            
            map<multimap<char, int>, int>::iterator it = a.find(temp);
            if(it != a.end())
                ret[(it->second)].push_back(strs[i]);
            else
            {
                a.insert(make_pair(temp,count));
                ret.push_back({strs[i]});
                count++;
            }
        }
        return ret;
    }
};

猜你喜欢

转载自blog.csdn.net/qq_31820761/article/details/91955027