[Python Algorithm] 자료구조(딕셔너리 해쉬) - 아나그램

> 자료구조 활용 - 스택, 큐, 해쉬, 힙 (해쉬 편)

 

 

📖 문제 : 아나그램

Anagram이란 두 문자열이 알파벳의 나열 순서를 다르지만 그 구성이 일치하면 두 단어는 아 나그램이라고 합니다.
예를 들면 AbaAeCe 와 baeeACA 는 알파벳을 나열 순서는 다르지만 그 구성을 살펴보면 A(2), a(1), b(1), C(1), e(2)로 알파벳과 그 개수가 모두 일치합니다. 즉 어느 한 단어를 재 배열하면 상대편 단어가 될 수 있는 것을 아나그램이라 합니다.

길이가 같은 두 개의 단어가 주어지면 두 단어가 아나그램인지 판별하는 프로그램을 작성하세 요. 아나그램 판별시 대소문자가 구분됩니다.

 

 

 

입력설명
첫 줄에 첫 번째 단어가 입력되고, 두 번째 줄에 두 번째 단어가 입력됩니다. 단어의 길이는 100을 넘지 않습니다.

 

출력설명
두 단어가 아나그램이면 “YES"를 출력하고, 아니면 ”NO"를 출력합니다.

 

입력예제 1
AbaAeCe baeeACA

 

출력예제 1

YES

 

import sys

a = input()
b = input()
str1 = dict()
str2 = dict()

for x in a:
	str1[x] = str1.get(x, 0) + 1
for x in b:
	str2[x] = str2.get(x, 0) + 1
    
for i in str1.keys():
    if i in str2.keys():
    	if str1[i] != str2[i]:
        	print("NO")
            break
    else:
    	print("NO")
        break
else:
	print("YES")

 

개선 코드 (딕셔너리 하나 쓰기)

import sys
a = input()
b = input()
dic = dict()

for x in a:
    dic[x] = dic.get(x,0) + 1
    
for x in b:
    dic[x] = dic.get(x,0) - 1
    
    
for x in a:
    if dic.get(x) > 0:
    	print("NO")
        break
        
else:
	print("YES")

⭐️ Point ! ⭐️

• str1[x] = str1.get(x, 0) + 1
  딕셔너리의 key값이 x인게 없으면 key값이 x이고, value값이 0인 원소를 넣고 / key값이 x인게 있으면 key값이 x인 value값을 +1 증가 시킴
  
  
• for i in str1.keys():
  딕셔너리 str1의 key값들 중에서 for문 이용해 값 찾기

 

 

출처

• 인프런 : 파이썬 알고리즘 문제 풀이