문자열 조작 : 그룹 애너그램(리트코드 49번)

▶ 문제 : https://leetcode.com/problems/group-anagrams/

Group Anagrams - LeetCode

##### 문제 #####
문자열 배열을 받아 애너그램 단위로 그룹핑하라.

##### 입력 #####
["eat", "tea", "tan", "ate", "nat", "bat"]

##### 출력 #####
[['eat', 'tea', 'ate'], ['tan', 'nat'], ['bat']]

 

▶ 접근 방법

애너그램이란 일종의 언어유희로 문자를 재배열하여 다른 뜻을 가진 단어로 바꾸는 것을 말한다.

애너그램을 판단하는 간단한 방법은 정렬하여 비교하는 것이다. 
애너그램 관계인 단어들을 정렬하면 서로 같은 값을 갖게 되기 때문이다.
아래 풀이에서는 정렬한 값을 키로하여 단어들을 딕셔너리에 추가한다.

딕셔너리에서 만약 존재하지 않는 키를 삽입하려 할 경우 KeyError가 발생하는데,
이를 방지하기 위해 defaultdict()로 딕셔너리를 선언하면
매번 키 존재 여부를 체크하지 않고 바로 값을 넣을 수 있다.

 

▶ 풀이 코드

from typing import List
from collections import defaultdict

ex_list = ["eat", "tea", "tan", "ate", "nat", "bat"]

def group_anagrams(strs: List[str]) -> List[List[str]]:
    anagrams = defaultdict(list)

    for word in strs:
        anagrams[''.join(sorted(word))].append(word)

    return list(anagrams.values())

print(group_anagrams(ex_list))


###################################################
a = ["cde", "cfc", "abc"]
# 정렬 기준을 문자열의 첫번째와 마지막 글자 기준으로 적용
print(sorted(a, key=lambda s: (s[0], s[-1])))

 

▶ 관련 링크

2021.12.14 - [알고리즘/리트코드] - 문자열 조작 : 유효한 팰린드롬(리트코드 125번)

2021.12.14 - [알고리즘/리트코드] - 문자열 조작 : 문자열 뒤집기(리트코드 344번)

2021.12.14 - [알고리즘/리트코드] - 문자열 조작 : 로그 파일 재정렬(리트코드 937번)

2021.12.14 - [알고리즘/리트코드] - 문자열 조작 : 가장 흔한 단어(리트코드 819번)