LeetCode 49. 字母異位詞分組

題目內容

給定一個字串陣列，將字母異位片語合在一起。字母異位詞指字母相同，但排列不同的字串。

示例:

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

解題思路

我的解法是使用輔助hashtable對單詞字母進行記錄與對比。
使用hashmap可能會好一點，不過我沒時間重寫。

解題程式碼

class Solution {
public:
    vector<vector<string>> groupAnagrams(vector<string>& strs) {
        vector<vector<string>> ans;
        vector<vector<int>> has;
        int i=0;
        while(i<strs.size())
        {
            if(i==0)
            {
                vector<string> t;
                t.push_back(strs[i]);
                ans.push_back(t);
                has.push_back(createH(strs[i]));
            }
            else
            {
                int j=0;
                vector<int> m;
                m=createH(strs[i]);
                while(j<ans.size())
                {
                    int k=0;
                    while(k<26)
                    {
                        if(m[k]!=has[j][k])break;k++;
                    }
                    if(k==26){ans[j].push_back(strs[i]);break;}
                    else j++;
                }
                if(j==ans.size()){
                    vector<string> t;
                    t.push_back(strs[i]);
                    ans.push_back(t);
                    has.push_back(m);
                }
            }
            i++;
        }
        return ans;
    }
    vector<int>createH(string &str)
    {
        vector<int> res(26);
        int i=0;
        while(i<str.size())
        {
            res[str[i]-'a']++;
            i++;
        }
        return res;
    }
};