백준 1009번 - 분산처리

 

https://www.acmicpc.net/problem/1009

1009번: 분산처리

 

 

 

 문제

재용이는 최신 컴퓨터 10대를 가지고 있다. 어느 날 재용이는 많은 데이터를 처리해야 될 일이 생겨서 각 컴퓨터에 1번부터 10번까지의 번호를 부여하고, 10대의 컴퓨터가 다음과 같은 방법으로 데이터들을 처리하기로 하였다.
1번 데이터는 1번 컴퓨터, 2번 데이터는 2번 컴퓨터, 3번 데이터는 3번 컴퓨터, ... ,
10번 데이터는 10번 컴퓨터, 11번 데이터는 1번 컴퓨터, 12번 데이터는 2번 컴퓨터, ...
총 데이터의 개수는 항상 a^b개의 형태로 주어진다. 재용이는 문득 마지막 데이터가 처리될 컴퓨터의 번호가 궁금해졌다. 이를 수행해주는 프로그램을 작성하라.

 입력

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

 출력

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

 예제 입력

5
1 6
3 7
6 2
7 100
9 635

 예제 출력

1
7
6
1
9

 

 

Solution

---

Python

# ver.1
T = int(input())
for i in range(T):
  a, b = map(int, input().split())
  val = pow(a, b, 10)

  if val == 0:
    print(10)
  else: 
    print(val)

# ver.2
p = [[2, 4, 8, 6], [3, 9, 7, 1], [4, 6], [], [], [7, 9, 3, 1], [8, 4, 2, 6], [9, 1]]

T = int(input())

for i in range(T):
    a, b = map(int, input().split())
    a %= 10
    if (0 < a <= 1 or 5 <= a <=6):
        print(a)
    elif(a == 0):
        print(10)
    elif(2 <= a <= 3 or 7 <= a <=8):
        print(p[a-2][b%4-1])
    elif(a == 4 or 9):
        print(p[a-2][b%2-1])

---