[백준 22251 🥇] 빌런 호석

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

22251번: 빌런 호석

문제풀이

import sys

# N층까지 존재
# K 자리 수 디스플레이 표현
# 최대 P개의 LED 반전
# 현재 X층에 멈춰있음
N,K,P,X=map(int, sys.stdin.readline().split(' '))

# LED를 배열로 표현
numbers=[ # 0 1 2 3 4 5 6 7 8 9
    [True, True,True,False,True,True,True],
    [False,False,True,False,False,True,False],
    [True, False,True,True,True,False,True],
    [True,False,True,True,False,True,True],
    [False, True,True,True,False,True,False],
    [True,True,False,True,False,True,True],
    [True,True,False,True,True,True,True],
    [True,False,True,False,False,True,False],
    [True,True,True,True,True,True,True],
    [True,True,True,True,False,True,True]
]

# 정수를 받아서 numbers 배열 형식으로 변환하는 함수
def change_to_arr(num):
    global numbers
    cur = []
    mod = num
    exp = K - 1
    while mod > 0:
        div = pow(10, exp)
        cur.append(numbers[mod // div])
        mod %= div
        exp -= 1
    # 0으로 끝나는 수 처리
    while exp>=0:
        cur.append(numbers[0])
        exp-=1
    return cur

# 두 배열의 차이의 개수를 구함
def diffNumber(arr1, arr2):
    diff=0
    for i in range(len(arr1)):
        for j in range(7):
            if arr1[i][j]!=arr2[i][j]:
                diff+=1
    return diff

ans=0
curX=change_to_arr(X)
for num in range(1, N+1):
    numArr=change_to_arr(num)
    diff=diffNumber(curX, numArr)
    if 1<=diff<=P:
        ans+=1

print(ans)

•  각 수별로 디스플레이의 상태를 numbers 배열에 저장한다.

• 나의 경우, 이런 식으로 각 수(0~9)를 인덱스별로 꺼져있는 부분(Fasle), 켜져있는 부분(True)으로 나타냈다.
•  change_to_arr 함수를 이용해서 정수형 숫자를 K 자리의 LED 배열로 변환했다.
• change_to_arr 함수에서 주의할 점은 500이나 410과 같은 마지막 수가 0인 숫자는 mod가 0보다 클 동안 반복되는 while문을 모든 자리 수를 배열로 바꾸지 못했을 때 빠져나오게 된다. 
• 그래서 exp가 0 이상인 상태로 첫 while문을 빠져나올 경우, exp가 0 이상일 동안 cur 배열에 numbers[0]을 추가해주고 exp를 -1해주는 while문을 추가했다. (첫번째 while 조건식을 exp가 0 이상일 때로 바꾸는 게 더 깔끔했을 것 같긴 하다 ㅎㅎ)
• 첫 번째로, 현재 층인 X를 change_to_arr 함수를 이용해서 배열로 바꾼 뒤에서 curX 변수에 저장했다. 
• 그리고 1부터 N까지의 수를 반복문을 돌면서 change_to_arr를 이용해서 배열로 바꾼 후에 numArr 변수에 저장했다.
• 각 반복문에서 diffNumber를 이용해서 curX와 numArr의 값이 다른 위치의 개수를 세고, 만약 이 다른 위치의 개수가 1 이상 P 이하인 수라면 변환이 가능한 경우이므로 ans+=1 해준다.
• ans를 출력하고 종료한다.