[Python] 백준 1009번 분산처리 | 런타임에러(KeyError)

분산처리 

문제

재용이는 최신 컴퓨터 10대를 가지고 있다. 어느 날 재용이는 많은 데이터를 처리해야 될 일이 생겨서 각 컴퓨터에 1번부터 10번까지의 번호를 부여하고, 10대의 컴퓨터가 다음과 같은 방법으로 데이터들을 처리하기로 하였다.

1번 데이터는 1번 컴퓨터, 2번 데이터는 2번 컴퓨터, 3번 데이터는 3번 컴퓨터, ... ,

10번 데이터는 10번 컴퓨터, 11번 데이터는 1번 컴퓨터, 12번 데이터는 2번 컴퓨터, ...

총 데이터의 개수는 항상 ab개의 형태로 주어진다. 재용이는 문득 마지막 데이터가 처리될 컴퓨터의 번호가 궁금해졌다. 이를 수행해주는 프로그램을 작성하라.

입력

입력의 첫 줄에는 테스트 케이스의 개수 T가 주어진다. 그 다음 줄부터 각각의 테스트 케이스에 대해 정수 a와 b가 주어진다. (1 ≤ a < 100, 1 ≤ b < 1,000,000)

출력

각 테스트 케이스에 대해 마지막 데이터가 처리되는 컴퓨터의 번호를 출력한다.

예제 입력 1 복사

5
1 6
3 7
6 2
7 100
9 635

예제 출력 1 복사

1
7
6
1
9

내 풀이

import sys
input = lambda: sys.stdin.readline().rstrip()
T = int(input())

com = {0:[10], 1:[1], 2:[2,4,8,6], 3:[3,9,7,1], 4:[4,6], 5:[5],
      6:[6], 7:[7,9,3,1], 8:[8,4,2,6], 9:[9,1]}

for _ in range(T):
    a, b = map(int, input().split())
    a = a%10
    c = b % len(com[a])
    print(com[a][c-1] if c != 0 else com[a][-1])

처음에 a**b % 10 이런 식으로 했더니 최대 100^1,000,000까지의 연산이 필요해지면서 시간초과가 뜸

제곱수의 일의 자리는 반복된다는 성질을 이용해서 풀었더니

런타임 에러 (KeyError) 🥹

찾아보니 해당되는 키가 없을 때 발생하는 에러라고 !

0을 추가해줬더니 됐다.

사실 dictionary 없이 리스트로 해도 될 듯