baekjoon - python - 1009

www.acmicpc.net/problem/1009

1009번: 분산처리

 

# @Author YoungMinKim
# baekjoon

import sys
N = int(sys.stdin.readline())
for _ in range(N):
    a,b = map(int,sys.stdin.readline().split())
    a = a%10
    if a == 0:
        print(10)
    elif a == 1 or a == 5 or a==6:
        print(a)
    elif a == 4 or a == 9:
        b = b%2
        if b == 1:
            print(a)
        else:
            print((a**2)%10)
    else:
        b = b%4
        if b == 0:
            print((a**4)%10)
        else:
            print((a**b)%10)
    

단순히 제곱해서 하면 당연히 시간초과

각 수를 제곱했을 때 1의 자리 변화 패턴을 이용하여 연산