Softeer H-클린알파 (난이도 4)

정답 코드

P, N = map(int, input().split())
A = list(map(int, input().split()))

ans = A[0]
for i in range(1, N):
    ans = (ans * P + A[i]) % 1000000007

print(ans)

후기

왜 난이도 4인 것인지 모르겠습니다. 난이도 1이 잘못 표시된 것 같습니다.