소프티어 | 바이러스

소프티어 | 바이러스

문제

바이러스가 숙주의 몸속에서 1초당 P배씩 증가한다.

처음에 바이러스 K마리가 있었다면 N초 후에는 총 몇 마리의 바이러스로 불어날까? N초 동안 죽는 바이러스는 없다고 가정한다.

제약조건

1 ≤ K ≤ 108인 정수

1 ≤ P ≤ 108인 정수

1 ≤ N ≤ 106인 정수

출력 형식

최종 바이러스 개수를 1000000007로 나눈 나머지를 출력하라.

내 풀이

import sys

K, P, N = map(int, input().split())

for i in range(N):
    K = (K*P)%1000000007

print(K)

---

풀이는 간단한 풀이이다. 알고리즘을 풀 때 print값을 result 로 뽑는 걸 좋아하고 습관이 되어있어서 아무생각없이 변수를 설정해주고 풀었는데 시간초과가났다.

import sys

K, P, N = map(int, input().split())
res = K

for i in range(N):
    res = res * P

result = res % 1000000007
print(result)

(위 내용과 같이 풀었다.)

무분별한 변수 설정,,에 대한 반성을 하게 되었다