[백준/2607] 비슷한 단어 (Python)

Problem : https://www.acmicpc.net/problem/2607

2607번: 비슷한 단어

 

Difficulty : Silver 3

 

Status : Solved

 

Time : 00:26:29

 

---

 

문제 설명

 

 

영문 알파벳 대문자로 이루어진 두 단어가 다음의 두 가지 조건을 만족하면 같은 구성을 갖는다고 말한다.

* 두 개의 단어가 같은 종류의 문자로 이루어져 있다.
* 같은 문자는 같은 개수 만큼 있다.

 

예를 들어 "DOG"와 "GOD"은 둘 다 'D', 'G', 'O' 세 종류의 문자로 이루어져 있으며 양쪽 모두 'D', 'G', 'O' 가 하나씩 있으므로 이 둘은 같은 구성을 갖는다. 하지만 "GOD"과 "GOOD"의 경우 "GOD"에는 'O'가 하나, "GOOD"에는 'O'가 두 개 있으므로 이 둘은 다른 구성을 갖는다.

두 단어가 같은 구성을 갖는 경우, 또는 한 단어에서 한 문자를 더하거나, 빼거나, 하나의 문자를 다른 문자로 바꾸어 나머지 한 단어와 같은 구성을 갖게 되는 경우에 이들 두 단어를 서로 비슷한 단어라고 한다.

예를 들어 "DOG"와 "GOD"은 같은 구성을 가지므로 이 둘은 비슷한 단어이다. 또한 "GOD"에서 'O'를 하나 추가하면 "GOOD" 과 같은 구성을 갖게 되므로 이 둘 또한 비슷한 단어이다. 하지만 "DOG"에서 하나의 문자를 더하거나, 빼거나, 바꾸어도 "DOLL"과 같은 구성이 되지는 않으므로 "DOG"과 "DOLL"은 비슷한 단어가 아니다.

입력으로 여러 개의 서로 다른 단어가 주어질 때, 첫 번째 단어와 비슷한 단어가 모두 몇 개인지 찾아 출력하는 프로그램을 작성하시오.

 

입력 및 출력

 

입력

첫째 줄에는 단어의 개수가 주어지고 둘째 줄부터는 한 줄에 하나씩 단어가 주어진다. 모든 단어는 영문 알파벳 대문자로 이루어져 있다. 단어의 개수는 100개 이하이며, 각 단어의 길이는 10 이하이다.

 

출력

입력으로 주어진 첫 번째 단어와 비슷한 단어가 몇 개인지 첫째 줄에 출력한다.

 

입력 예시

 

4
DOG
GOD
GOOD
DOLL

 

출력 예시

 

2

 

 

---

 

풀이

 

두 문자열의 문자를 카운팅한다고 생각하자. (문제 조건상 순서가 중요하지 않기 때문이다) 즉 딕셔너리 구조를 사용해서 접근한다고 가정하자.

 

딕셔너리의 두 문자열의 모든 원소를 비교해본다면, 다음과 같은 상황 세 가지가 가능하다.

 

• 두 문자열의 길이가 같고, 구성까지 동일한 경우 : 이 경우 모든 딕셔너리의 원소 카운팅 개수 역시 동일하므로 다른 점은 0이 되며, 길이상으로도 동일하다.
• 두 문자열의 길이가 같고, 구성이 정확히 2개가 차이나는 경우 : 이 경우 단 하나의 문자열이 차이가 나므로 정확히 구성이 2개가 차이가 나야 한다. 즉 두 딕셔너리의 카운팅 키는 정확히 2개가 차이난다.
• 두 문자열의 길이가 1 차이가 나고, 구성도 1 차이나는 경우 : 이 경우는 두 문자열에서 하나를 삭제하거나 혹은 다른 한 문자열에서 하나를 추가함으로써 동일구조로 만들어 줄 수 있는 케이스이다.

 

즉 세 가지 경우에 한하여 정답을 카운팅하면 된다.

 

 

풀이 코드

from collections import defaultdict
import sys
input = sys.stdin.readline

N = int(input())

def make_dict(word) :
  length = len(word)
  result = defaultdict(int)
  for s in word :
    result[s] += 1
  return result, length

def compare_dict(target) :
  global word_dict, str_set
  diff = 0
  union_set = str_set.union(set(target.keys()))
  for s in union_set :
    diff += abs(word_dict[s] - target[s])
  return diff

answer = 0
word_dict, word_len = make_dict(input().strip())
str_set = set(word_dict.keys())
for _ in range(N-1) :
  target_dict, target_len = make_dict(input().strip())
  diff = compare_dict(target_dict)
  if (diff == 2 or diff == 0) and word_len == target_len :
    answer += 1
  elif diff == 1 and abs(word_len - target_len) == 1 :
    answer += 1

print(answer)

 

풀이 완료!