Jae-Kyung Cho LLM Developers who was a Robotics engineer

Baekjoon-11066-파일 합치기

마찬가지로 PyPy3로 풀어야 하는 문제… 그럼 만약에 PyPy3가 아니라 Python3로 채점한다고 하면 꼼짝없이 C++로 코딩해야 하는건가?

import sys
from math import factorial as fac
input = sys.stdin.readline

T = int(input())

result = []
for _ in range(T):
    K = int(input())
    sizes = list(map(int,input().split()))

    dp = [[int(1e9)]*K for _ in range(K)]
    for i in range(K):
        dp[i][i] = 0
    for i in range(1,K):
        for j in range(K-i):
            merge_size = sum(sizes[j:i+j+1])
            for k in range(i):
                dp[j][j+i] = min(dp[j][j+i], merge_size + dp[j][j+k]+dp[j+k+1][j+i])
    result.append(dp[0][K-1])

for r in result:
    print(r)
Comments