动态规划 石子合并问题伪代码

以下是石子合并问题的动态规划伪代码：

function mergeStones(stones, K):
    n = length(stones)
    if (n == 1) return 0
    if ((n - 1) % (K - 1) != 0) return -1
    
    prefixSum = [0] * (n + 1)
    for i = 1 to n:
        prefixSum[i] = prefixSum[i - 1] + stones[i - 1]
    
    dp = [[0] * (n + 1) for _ in range(n + 1)]
    for l = K to n:
        for i = 1 to n - l + 1:
            j = i + l - 1
            dp[i][j] = float('inf')
            for m = i + K - 1 to j step K - 1:
                dp[i][j] = min(dp[i][j], dp[i][m] + dp[m + 1][j])
            if ((l - 1) % (K - 1) == 0):
                dp[i][j] += prefixSum[j] - prefixSum[i - 1]
    
    return dp[1][n]

其中，`stones` 是一个长度为 `n` 的数组，表示石子的重量；`K` 是一个正整数，表示每次合并的石子堆数。函数返回值为石子合并的最小代价，如果无法合并则返回 `-1`。