알고스팟 hamming code 문제 파이썬으로 풀기

문제 링크: https://algospot.com/judge/problem/read/HAMMINGCODE

암호화된 코드를 받아 암호화 되기 전의 코드를 출력하면 된다.

 

#hammingcode
import sys
 
 
case = int(sys.stdin.readline())
for i in range(case):
    en = sys.stdin.readline()  # 코드 입력
    encode = []
    for j in range(7):  # 전체를 받으면 엔터키까지 받아져서 코드 크기만큼만 써야 함
        encode.append(int(en[j]))  # 코드 숫자화
    syn0 = encode[0]^encode[2]^encode[4]^encode[6]  # parity bit 구하기
    syn1 = encode[1]^encode[2]^encode[5]^encode[6]
    syn3 = encode[3]^encode[4]^encode[5]^encode[6]
    if syn0+syn1+syn3 != 0:  # parity bit 들의 합이 0이 아니면, 즉 에러가 생겼다면
        syndrome = str(syn3) + str(syn1) + str(syn0)  # syndrome 구하기
        error = int(syndrome, 2)-1  # 문제가 있는 비트의 위치 찾기
        if encode[error] == 0:  # 비트 바꾸기
            encode[error] = 1
        else:
            encode[error] = 0
    result = str(encode[2]) + str(encode[4]) + str(encode[5]) + str(encode[6])  # 결과 출력
    print(result)
 

코드는 이렇다.