[Softeer][level2][21년 재직자 대회 예선] 전광판 - 파이썬(Python) 풀이

https://softeer.ai/practice/6268

Softeer - 현대자동차그룹 SW인재확보플랫폼

현대차그룹에 다니는 당신은 전세계 유가 변동에 대해 실시간으로 파악하기 위해 사무실에 유가를 실시간으로 표시하는 전광판을 설치하였다. 전광판은 최대 다섯 자리의 자연수만을 표시할 수 있도록, 아래와 같이 육각형 모양의 전구 7×5=35개로 구성되어 있다.

 

 

8자 모양의 전구 묶음은 0부터 9까지의 숫자를 표현할 수 있으며, 표현 방법은 아래와 같다. 아래 그림에서 전구가 켜졌으면 검정색, 꺼졌으면 옅은 회색으로 표현되었다.

 

 

예를 들어, 전광판을 통해 9881를 표현하면 아래와 같다. 만의 자리 수가 없기 때문에, 만의 자리에 해당하는 전구들이 모두 꺼져 있음에 유의하라.

 

 

예를 들어, 전광판을 통해 10724를 표현하면 아래와 같다.

 

 

각각의 전구에는 스위치가 달려 있다. 전구에 달려 있는 스위치를 누를 때, 그 전구가 켜져 있었다면 꺼지고, 그 전구가 꺼져 있었다면 켜진다.

 

지금 전광판에 자연수 A가 표시되어 있는데, 유가가 변동됨에 따라 전광판에 표시된 자연수를 B로 바꿔야 한다. 이러한 목표를 달성하기 위해 스위치를 최소 몇 번 눌러야 하는지를 구하는 프로그램을 작성하라.

---

- 문제 풀이 과정

처음에는 숫자 차이 경우의 수 모두를 구하려다가 불빛이 7개라는 말을 보고 힌트를 얻어서 접근했다.

다만 두번째 for loop가 지저분해서 나 자체도 여기서 머리가 좀 아팠다....

만약 다시 한다면, 해설과 같이 for 루프를 추가해서

answer += (num[lst_a[i]][j] != num[lst_b[i]][j])

이런 방식으로 접근을 해야겠다.

- 내 코드

import sys

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

t = int(input())

for _ in range(t):
    answer = 0
    lst_a, lst_b = map(str, input().split())
    
    lst_a = [x if x.isdigit() else 'space' for x in lst_a]
    lst_b = [x if x.isdigit() else 'space' for x in lst_b]

    lst_a = ['space'] * (5 - len(lst_a)) + lst_a
    lst_b = ['space'] * (5 - len(lst_b)) + lst_b

    for i in range(5):
        a = [int(x) for x in list(num[str(lst_a[i])])]
        b = [int(x) for x in list(num[str(lst_b[i])])]
        result_list = [x + y for x, y in zip(a, b)]
        answer += result_list.count(1)
    print(answer)