leetcode 49. 字母异位词分组

问题描述

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

示例:

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

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

问题分析

用质数表示26个字母，把字符串的各个字母相乘，这样可保证字母异位词的乘积必定是相等的。注意的是乘积如果用int,long,float都会造成溢出，因此使用double.

代码

class Solution {
public:
    vector<vector<string>> groupAnagrams(vector<string>& strs) {
        int prime[26] = { 2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 41, 43, 47, 53, 59, 61, 67, 71, 73, 79, 83, 89, 97, 101, 103 };//26个质数
        int n = strs.size(),k=0;
        //float prod;
        double prod;
        vector<vector<string>> ans;
        if(n == 0)return ans;
        map<double, int> group;
        for(int i = 0; i < n; i++)
        {
            int m = strs[i].size();
            prod = 1;
            for(int j = 0; j < m; j++)prod *= prime[strs[i][j]-'a'];
            if(group.find(prod) == group.end())
            //if(!group.count(prod))
            {
                group[prod] = k++;
                vector<string> tmp;
                tmp.push_back(strs[i]);
                ans.push_back(tmp);
            }
            else{
                ans[group[prod]].push_back(strs[i]);
            }
        }
        return ans;
    }
};

结果：

执行用时 :28 ms, 在所有 C++ 提交中击败了99.49%的用户
内存消耗 :19.4 MB, 在所有 C++ 提交中击败了42.63%的用户