백준(boj) 1009 파이썬 - 분산처리

문제 링크 : https://www.acmicpc.net/problem/1009

1009번: 분산처리

 

단순히 문제해결을 하면 간단하지만, 시간초과가 자꾸 발생하여 여러번 고쳐야했다.

1의 자릿수만 생각을 하면 간단한 문제였다.

1, 5, 6은 몇제곱을 해도 그대로 출력하면 된다.

나머지 2,3,7,8은 제곱이 거듭됨에 따라 4가지 규칙에 의해 수가 달라진다.

9와 4는 2가지 규칙이 있었다.

(일의자릿수만 따졌을때) 2의 거듭제곱 : 2, 4 , 8 , 6 .... 반복  3의 거듭제곱 : 3, 9,  7,  1 .... 반복 나머지 수들도 마찬가지 형식

 

 

T = int(input())
for _ in range(T):
    a, b = map(str, input().split())
    b = int(b)
    if a[-1]=='1' or a[-1]=='5' or a[-1]=='6' : print(a[-1])   

    elif a[-1]=='2' or a[-1]=='3' or a[-1]=='7' or a[-1]=='8' :
        if b%4==0 : print((int(a[-1])**4)%10)
        elif b%4 == 1 : print(a[-1])
        elif b%4 == 2  : print((int(a[-1])**2)%10)
        elif b%4 == 3 : print((int(a[-1])**3)%10)

    elif a[-1]=='9' or a[-1]=='4':
        if b%2 == 0 : print((int(a[-1])**2)%10)
        elif b%2 == 1: print(a[-1])

    elif a[-1] == '0' : print(10)