Python3/백준 알고리즘

2022.02.10 [백준] (python 파이썬) 분산처리

ian's coding 2022. 2. 10. 19:00
728x90
반응형

 

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

 

1009번: 분산처리

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

www.acmicpc.net

 

풀이

 

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
import sys
 
n=int(sys.stdin.readline())
arr_a=[]
arr_b=[]
arr_ans=[]
for i in range(n):
    a,b=map(int, sys.stdin.readline().split())
    arr_a.append(a%10)
    arr_b.append(b)
for i in range(n):
    if arr_a[i]==0:
        arr_ans.append(10)
    elif arr_a[i]==1 or arr_a[i]==5 or arr_a[i]==6:
        arr_ans.append(arr_a[i])
    elif arr_a[i]==4 or arr_a[i]==9:
        if arr_b[i]%2==0:
            arr_ans.append(arr_a[i] **2 % 10)
        else:
            arr_ans.append(arr_a[i])
 
    else:
        if arr_b[i]%4==0:
            arr_ans.append(arr_a[i]**4%10)
        else:
            arr_ans.append(arr_a[i]**(arr_b[i]%4)%10)
for i in arr_ans:
    print(i)
cs

 

 

 

 

728x90
반응형