코테 공부/python

[소프티어] #624 Python 전광판

sweet-po 2023. 8. 7. 11:16

https://softeer.ai/practice/info.do?idx=1&eid=624

처음에 스터디원이 껏다 키는 전구 자체 위치로 이진수 만들어서 하는 아이디어 안줬으면 못풀었다ㅠ

항상 고맙고 고마운 나의 말랑이 스터디원..

0과 빈 자리수 전구 세팅 다르단것도 풀면서 결과가 제대로 안나와서 알아채고 빈자리수 표시인 x를 딕셔너리에 추가로 넣어줬다.

#https://softeer.ai/practice/info.do?idx=1&eid=624

import sys
input = sys.stdin.readline

n = int(input())
li = []
for _ in range(n):
    temp1, temp2 = map(str, input().split())
    li.append([temp1, temp2])

dic = {'x':'0000000', '0':'1110111', '1':'0010010', '2':'1011101', '3':'1011011', '4':'0111010', '5':'1101011', '6':'1101111', '7':'1110010', '8':'1111111', '9':'1111011'}

for i in range(n):
    A, B = li[i][0], li[i][1]
    cnt = 0
    #자리수 맞추기
    maxlen = max(len(A), len(B))
    if len(A) < maxlen:
        A = 'x'*(maxlen-len(A)) + A
    else:
        B = 'x'*(maxlen-len(B)) + B

    for l in range(maxlen):
        digitA = dic.get(A[l])
        digitB = dic.get(B[l])
        for m in range(7):
            if digitA[m] != digitB[m]:
                cnt += 1

    print(cnt)