BOJ/문자열 (해시,맵)

[Python] 백준 - 6996번 : 애너그램

JWonK 2021. 12. 24. 00:37
728x90
반응형

https://www.acmicpc.net/problem/6996

 

6996번: 애너그램

첫째 줄에 테스트 케이스의 개수(<100)가 주어진다. 각 테스트 케이스는 한 줄로 이루어져 있고, 길이가 100을 넘지 않는 단어가 공백으로 구분되어서 주어진다. 단어는 알파벳 소문자로만 이루어

www.acmicpc.net

문제

두 단어 A와 B가 주어졌을 때, A에 속하는 알파벳의 순서를 바꾸어서 B를 만들 수 있다면, A와 B를 애너그램이라고 한다.

두 단어가 애너그램인지 아닌지 구하는 프로그램을 작성하시오.

 

두 단어가 주어졌을 때 애너그램인지 아닌지만 확인하면 되는 간단한 문제이다.

 

1. 정렬을 이용하여 첫 번째 단어를 알파벳 순으로 정렬해준 후 딕셔너리에 넣어준다

2. 두 번째 단어를 1번과 같이 정렬을 이용하여 알파벳 순으로 정렬해준다

3. 정렬한 두 번째 단어가 딕셔너리 안에 존재하면 애너그림이고, 존재하지 않는다면 애너그램이 아닌 것으로 판단한다.

 

위 과정을 코드로 작성하면 된다

import sys
from collections import defaultdict, Counter

def isanagram(strs:list):
    anagram = defaultdict()
    anagram[''.join(sorted(strs[0]))] = 1

    if ''.join(sorted(strs[1])) in anagram:
        return True
    else:
        return False

N = int(sys.stdin.readline())
for _ in range(N):
    strs = list((sys.stdin.readline().split()))
    if isanagram(strs) == True:
        print(f'{strs[0]} & {strs[1]} are anagrams.')
    else:
        print(f'{strs[0]} & {strs[1]} are NOT anagrams.')
728x90
반응형