Softeer 지도 자동 구축 (난이도 2)

정답 코드

N = int(input())

dp = [0 for i in range(16)]
dp[0] = 4
dp[1] = 9
for i in range(2, N + 1):
    s = 4 ** (i - 1)
    root = s ** (1/2)
    dp[i] = dp[i - 1] + 5 * s - 2 * root * (root - 1)

print(int(dp[N]))

후기

기본 구현