알고리즘/문제풀이

구현(비슷한 단어/S3) - 2067번

문승주 2022. 11. 30. 12:23
반응형

1. 문제

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

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

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

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

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

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

 

2. 입력

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

4
DOG
GOD
GOOD
DOLL

 

3. 출력

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

2

 

4. 코드

import sys

n = int(sys.stdin.readline())

words = [sys.stdin.readline().rstrip() for _ in range(n)]

# 첫 단어를 리스트로 변환하여 대입
first = list(words[0])
result = 0

for i in range(1, n):
  # 단어를 리스트로 변환
  chars = list(words[i])
  count = 0
  r = False
  # chars의 길이가 first보다 1이 작거나 같을때
  if len(chars) == len(first) or len(chars) == len(first) + 1:
    x = list(chars)
    for f in first:
      if f in x:
        x.remove(f)
        count += 1
    if len(chars) - 1 <= count:
      r = True
  # chars의 길이가 first보다 1이 클 때
  elif len(chars) == len(first) - 1:
    y = list(first)
    for c in chars:
      if c in y:
        y.remove(c)
        count += 1
    if len(chars) == count:
      r = True
  else:
    continue
  # r이 참이면 결과값을 추가
  if r:
    result += 1
print(result)

 

 

5. 풀이설명

① 입력값 n가 단어리스트 words를 입력받는다.

 

② 두번째 줄에 받은 첫단어를 리스트로 변환하여 first에 대입한다.

 

③ 세번째 줄부터 시작하는 반복문을 실행한다.

 

④ 단어 words[i] 를 리스트로 변환하여 chars에 대입하고 count를 0으로, r 을 False로 선언한다.

 

- chars 길이가 ②에서 대입한 first 길이보다 1이 작거나 같은 경우

  a. chars를 x에 대입하고 x안에 first가 존재하면 count를 1 추가한다.

  b. count가 chars의 길이보다 크거나 같으면 r을 True로 변환

 

- chars 길이가 ②에서 대입한 first 길이보다 1이 클 경우

  a. first를 y에 대입하고 y안에 chars존재하면 count를 1 추가한다.

  b. count가 chars의 길이와 같으면 r을 True로 변환

 

- 둘다 아닐 경우

  continue로 다음 반복문을 실행한다.

 

⑥ r이 True이면 결과값에 1을 추가한다.

 

⑦ 결과값을 출력한다.

 

6. 느낀점

파이썬에서 리스트를 할당할때는 list로 감싸주거나 copy 함수를 사용해야되는 것을 알게되었다.

반응형