[백준] 분산처리 / 1009번 / 파이썬 / python / 구현

📌 문제

재용이는 최신 컴퓨터 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)

💬 출력 

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

 

---

👩‍💻 code

import sys 
input = sys.stdin.readline

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

	if aa == 0: # 패턴 1
		print(10)
	elif aa in [1,5,6]: 
		print(aa)
	elif aa in [4,9]: #패턴 2
		bb=b%2
		if bb == 0:
			print(aa*aa%10)
		else:
			print(aa)
	else: #패턴 4
		bb=b%4  
		if bb ==0:
			print(aa**4%10)
		else:
			print(aa**bb%10)

---

💕 해설

단순히 a**b를 사용해서 일의자리 수를 출력하는건 시간 초과가 뜬다. ex) 7**100

문제에서 준 컴퓨터가 10대라는 것이 큰 힌트인데, 

1 부터 10 까지 각각을 거듭제곱했을 시 일의자리 들의 패턴을 정리해보면 아래 표와 같다.

1	1
2	2, 4, 6, 8
3	3, 9, 7, 1
4	4, 6
5	5
6	6
7	7, 9, 3, 1
8	8, 4, 2, 6
9	9, 1
10	0

1. 패턴이 하나

1,5,6은 자기자신을 그대로 출력하게끔 하고, 10일때는 0이므로 10(컴퓨터번호니까) 출력하도록 한다.

 

2. 패턴이 2개

이번엔 b도 10으로 나눈 나머지를 구하고 , 나머지가 0이면 a를 그대로 출력, 아니라면 a * a 의 일의자리를 출력.

 

3. 패턴이 4개

b를 10으로 나눈 나머지를 bb라고 하고, 나머지가 0이면 a를 그대로 출력, 아니라면 a를 bb만큼 거듭제곱한것의 나머지를 출력하도록 한다.

 

# 처음 풀이 - 너무나도 당연한 시간 초과,,,
import sys 
input = sys.stdin.readline
t = int(input())
for _ in range(t):
	a,b = map(int,input().split())
	c = str(a**b)
	print(c[-1])