[LeetCode] 49. Group Anagrams - Python

🤔 문제

---

 

Group Anagrams - LeetCode

 

😀 풀이

---

위 문제는 주어진 문자열 배열들을 anagram 끼리 모아보는 문제입니다.

 

문제 풀이에서 핵심이 된 아이디어는 "anagram이라면 문자열을 정렬했을 경우 같은 값이 나온다" 입니다.

 

따라서 풀이를 진행한 흐름은 아래와 같습니다.

1. 문자열 배열을 하나씩 살펴보는 과정에서 각 문자열을 정렬해본다.
2. 정렬한 뒤 dict에 문자열을 넣는다.
3. 이 때 key는 정렬된 문자열, value는 원래 문자열이다. ( value는 배열에 저장됨 )
4. 문자열 배열을 다 살펴보았다면 dict의 values들이 문제에서 요구하는 정답이 됩니다.

이 때 2 ~ 3번 과정을 간단히 하기위해 파이썬에서 제공하는 defaultdict를 사용하면 편리합니다.

---

💻 풀이 코드는 아래와 같습니다.

from collections import defaultdict

class Solution:
    def groupAnagrams(self, strs: List[str]) -> List[List[str]]:
        anagrams = defaultdict(list)

        for str in strs:
            sorted_str = ''.join(sorted(str))
            anagrams[sorted_str].append(str)

        return anagrams.values()