Leetcode - 49. Group Anagrams
August 07, 2021
문제
문자열 배열을 받아 애너그램 단위로 그룹화 하라
input
strs = ["eat","tea","tan","ate","nat","bat"]output
[["bat"],["nat","tan"],["ate","eat","tea"]]나의 풀이
import collections
class Solution:
def groupAnagrams(self, strs: List[str]) -> List[List[str]]:
anagrams = collections.defaultdict(list)
for word in strs:
key = ''.join(sorted(word))
anagrams[key].append(word)
return anagrams.values()인자로 주어진 strs에서 하나 꺼내 정렬한다. 정렬을 수행하면 ate, eat모두 모두 같은 문자로 이뤄져 있으므로 모두 ['a', 'e', 't']로 정렬이 된다. 이 리스트를 키로 삼는 dictionary자료형인 anagrams를 하나 생성한다.
이때, 없는 키를 생성하려 할 경우 KeyError가 발생하므로defaultdict()로 선언해준다. 이제 정렬된 key를 dictionary의 key로 삼고 어를 하나씩 list에 삽입시켜주고 anagrams의 값만 리턴시켜준다.
