LeetCode : 49. Group Anagrams

String을 다루는 문제였다.

 

[메인 로직]

map을 사용하여 각각의 문자열을 정렬한 결과와 같은 문자열들을 하나의 vector에 넣어 분류해주었다.

aet -> ["eat", "tea", "ate"]

ant -> ["tan", "nat"]

...

 

class Solution {
public:
    map<string, int> m;
    vector<vector<string>> ans;
    vector<vector<string>> groupAnagrams(vector<string>& strs) {
        ans.resize(strs.size());
        int idx = 0;
        for(int i=0; i<strs.size(); i++){
            string temp = strs[i];
            sort(temp.begin(), temp.end());
            if(m.count(temp)==0){
                m[temp] = idx++;
                ans[m[temp]].push_back(strs[i]);
            }else{
                ans[m[temp]].push_back(strs[i]);
            }
        }
        
        for(int i=0; i<ans.size(); i++){
            if(ans[i].size()==0){
                ans.erase(ans.begin()+i, ans.end());
            }
        }
        return ans;
    }
};