[백준 - 구현] 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)

 

출력

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

 

💡문제 해결 아이디어

• 컴퓨터가 1번부터 10번까지 번호가 부여된 것이 키 포인트
• a의 1의 자리에 따라 n제곱 했을 경우 1의 자리 패턴이 정해져 있음
  
  • 1, 5, 6 : 그대로 출력. 10의 경우 0이므로 10(컴퓨터번호) 출력 
  • 4, 9 : b를 2로 나누었을 때 나머지가 1일 경우 그대로 출력, 0일 경우 a^2의 1의자리 출력
  • 2, 3, 7, 8: b를 4로 나누어 봤을 때 나머지가 0일 경우 a^4의 1의 자리 출력, 아닐경우 (a의 1의자리) x (b의 나머지)의 1의 자리 출력

# 시간초과 코드
t = int(input())

for _ in range(t):
    a, b = map(int, input().split())
    print((a ** b) % 10)

t = int(input())

for _ in range(t):
    a, b = map(int, input().split())
    aa = a % 10 # a의 1의 자리
    
    # a의 1의자리가 0이라면(a는 10의 배수)
    if aa ==0: 
        print(10)
        
    # a의 1의자리가 1, 5, 6 이라면
    elif aa in [1, 5, 6]:
        print(aa) # 그대로 출력
    
    # a의 1의자리가 4, 9 라면
    elif a in [4, 9]:
        if b % 2 == 0: # 2가지 경우 존재
            print((aa ** 2) % 10)
        else:
            print(aa)

	# a의 1의자리가 2, 3, 7, 8 이라면
	else:
        bb = b % 4 # 4가지 경우 존재
        if bb == 0:
            print((aa ** 4) % 10)
        else:
            print((aa ** bb) % 10)