[python] Level2_전광판

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

Softeer

 

import sys
#sys.stdin = open("input2.txt",'r')
 
#테스트 케이스
T = int(sys.stdin.readline().rstrip())
 
# 0~9까지 미리 켜져야 하는 라이트 정해놓기
light_dict = {}
#라이트 구역은 0~6개로 총 7개
light_dict[0]=[1,1,1,1,1,1,0]
light_dict[1]=[0,0,0,0,1,1,0]
light_dict[2]=[1,0,1,1,0,1,1]
light_dict[3]=[1,0,0,1,1,1,1]
light_dict[4]=[0,1,0,0,1,1,1]
light_dict[5]=[1,1,0,1,1,0,1]
light_dict[6]=[1,1,1,1,1,0,1]
light_dict[7]=[1,1,0,0,1,1,0]
light_dict[8]=[1,1,1,1,1,1,1]
light_dict[9]=[1,1,0,1,1,1,1]
light_dict["empty"]=[0,0,0,0,0,0,0]
 
for i in range(T):
    #빈전광판
    array_left=[]
    array_right=[]
    n,m = map(int, sys.stdin.readline().split())
    
    #n,m 비교하기 쉽게 분리하기
    array_left = list(str(n))
    array_right = list(str(m))
    for i in range(len(array_left)):
        array_left[i] = int(array_left[i])
    for i in range(len(array_right)):
        array_right[i] = int(array_right[i])
    
    #갯수가 다르면 empty채우기
    if len(array_left) > len(array_right):
        for i in range(len(array_left)-len(array_right)):
            array_right.insert(0,"empty")
    elif len(array_left) < len(array_right):
        for i in range(len(array_right)-len(array_left)):
            array_left.insert(0,"empty")
 
    #마지막순서 비교하기
    answer = 0
    for i in range(len(array_left)):
        #값이 다를 경우 다른만큼 더하기
        if light_dict[array_left[i]] != light_dict[array_right[i]]:
            for j in range(7):
                if light_dict[array_left[i]][j] != light_dict[array_right[i]][j]:
                    answer +=1
    print(answer)