[Python] 백준 2596번 비밀편지

문제

병현이는 지은이에게 문자 A, B, C, D, E, F, G, H 로 쓰여진 편지를 날마다 보내는데, 컴퓨터로 보내는 비밀편지로, 한 문자마다 0 또는 1인 숫자 여섯 개를 사용하여 보낸다. 둘 사이의 약속은 다음과 같다.
A 000000
B 001111
C 010011
D 011100
E 100110
F 101001
G 110101
H 111010

병현이가 어느 날 001111000000011100 을 보내면 지은이는 이것을 BAD로 이해하게 된다. 그런데 둘 사이에 약속이 잘 만들어져 있기 때문에, 통신에 문제가 생겨서 한 문자를 표시하는 여섯 숫자 중 어느 한 숫자만 틀리게 오는 경우, 지은이는 원래 보내려는 문자를 알아 낼 수가 있다.
예를 들어 지은이가 000100을 받았을 때, A와 숫자 한자만 다르고, 다른 문자들과는 각각 숫자 두 자 이상이 다르므로 지은이는 이것이 A라고 알아보게 된다.
다만 111111과 같이 모든 문자의 표현과 숫자 두 자 이상이 다른 경우에는 무슨 문자인지 알 수가 없게 된다. 예를 들어 지은이가 011111000000111111000000111111 을 받았을 때, BA 다음에 알아 볼 수 없는 문자가 나오는데. 이 경우 이런 것이 처음 나오는 문자의 위치인 3을 출력한다.
지은이가 받은 편지를 보고 문자들을 알아내어 출력하거나, 모르는 문자가 있는 경우, 이것이 처음 나오는 위치를 출력하는 프로그램을 작성하시오.

 

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

입력

첫줄에는 보낸 문자의 개수(10개 보다 작다.)가 입력된다. 다음 줄에는 문자의 개수의 여섯 배 만큼의 숫자 입력이 주어진다.

출력

주어진 입력에서 지은이가 이해한 문자들을 출력하거나, 모르는 문자가 나오는 경우 그런 것이 처음 나오는 위치를 출력한다.

예제

# input
3
001111000000011100
# output
BAD

# input
5
011111000000111111000000111111
# output
3

문제 풀이

n = int(input())
S = input()
D = {'000000' : 'A', '001111' : 'B', '010011' : 'C', '011100' : 'D', 
     '100110' : 'E', '101001' : 'F', '110101' : 'G', '111010' : 'H'}
answer = []

for i in range(0, n*6, 6):
    code = S[i:i+6]
    
    # 딕셔너리에 현재 코드가 존재하면 매핑된 문자를 추가
    if code in D:
        answer.append(D[code])
    else:
        # 현재 코드와 해밍 거리 1인 문자 후보를 저장할 리스
        possible = []
        
        # 모든 키와 비교하여 해밍 거리 계산
        for key, val in D.items():
            # 해밍 거리가 1인 경우 후보 리스트에 추가
            if sum(code1 != code2 for code1, code2 in zip(code, key)) == 1:
                possible.append(val)
          
        # 후보가 하나뿐이라면 해당 문자를 추가
        if len(possible) == 1:
            answer.append(possible[0])
        # 후보가 없거나 여러 개인 경우, 실패한 문자의 위치 출력 후 종료
        else:
            print(i // 6 + 1)
            break
        
# 모든 문자를 성공적으로 해독한 경우 결과 출력
else:
    print(''.join(answer))

---

파이썬을 독학하시는 분들에게 도움이 되길 바라며,

혹 더 좋은 방법이 있거나 오류가 있다면 편하게 말씀 부탁드립니다.