[구현] 백준 1009: 분산처리

문제:재용이는 최신 컴퓨터 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)

 

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

 

풀이 코드

def find_last_num(a, b):
    ans_list = []
    c = a
    a = a % 10
    while True:
        if a in ans_list:
            break
        else:
            ans_list.append(a)
            a = (a * c) % 10

    # b가 1이면 첫번째, 2면 2번째... 10이고 리스트 길이가 6이면
    # [1, 2, 3, 4, 5, 6]
    # print(ans_list)
    if ans_list[b % len(ans_list) - 1] == 0:
        return 10
    return ans_list[b % len(ans_list) - 1]

t = int(input())
for i in range(t):
    a, b = map(int, input().split())
    print(find_last_num(a, b))

 

a**b를 계산해서 1의자리를 return하는 방식은 시간초과