题目

思路

话不多说，直接上代码

附上INT_MAX和INT_MIN

【C++】详解 INT_MAX 和 INT_MIN（INT_MAX 和 INT_MIN是什么？它们的用途是什么？如何防止溢出？）_c++ int max-CSDN博客

代码

/*
leetcode合并石头的最低成本-XMUOJ元素共鸣：深层次的唤醒
--JinlongW-2024/05/26
*/
/*
定义dp[i][j][k]为合并第i堆到第j堆石头为k堆的成本，状态转移方程有关键两点：1.dp[i][j][1] = dp[i][j][k] + sum(i, j)。不能直接求出合并为1堆的成本，得先合并成k堆。
2.dp[i][j][m] = min(dp[i][p][1] + dp[p + 1][j][m - 1])，i <= p < j，2 <= m <= k。
这里m为堆数，不能直接用k是因为右部分的存在，要对右部分继续划分求解的话，子问题就必须有合并成m堆的情况。
细节：
第一点：一定会有不能合并成1堆的情况，怎么排除掉这种情况呢？如果能合并成1堆，就一定得先合并成k堆，这在前面已经讨论过了。这k堆里面的其中1堆，是由k堆合并而来的，这样一直套娃，就能还原到原始的堆数n。我们由此可以定义一个方程：k + (k - 1) * a == n，a是一个大于等于0的整数。推算一下，有：k - 1 + (k - 1) * a == n - 1 ==》 (k - 1) * (a + 1) == n - 1。所以对于有解的情况，一定有(n - 1) % (k - 1) == 0。第二点：枚举分界点时，step应该是k - 1而不是1。step为1当然也是正确的，但是却进行了很多无用的计算，导致运行时间增加。为什么step可以是k-1呢？因为我们设计的划分是将左部分区间[i, p]合并为1堆，那就一定有(p - i) % (k - 1) == 0，结合最初p = i，就可以知道step应该是k-1，这样会涵盖所有有效的分界点p。对于其他的分界点p，左部分不能合并为1堆，那这样的划分并没有意义，对于计算答案也就没有帮助了。第三点： 最大数INT_MAX,最小数INT_MIN 感悟：本题起初的时候有几个点没过就是因为max设太小，只有0x3f,导致后面两个数据点没过。 
*/
#include<iostream>
#include<vector>
#include<algorithm>
#include<climits>
using namespace std;const int N = 35;
int n, k;
int a[N], S[N];
int dp[N][N][N];int main() {cin >> n;for (int i = 1; i <= n; ++i) {cin >> a[i];S[i] = S[i - 1] + a[i];}cin >> k;if ((n - 1) % (k - 1) != 0) {cout << -1 << endl;return 0;}for (int len = 2; len <= n; ++len) {for (int i = 1; i + len - 1 <= n; ++i) {int j = i + len - 1;for (int m = 2; m <= k; ++m) {dp[i][j][m] = INT_MAX;//初始化 for (int p = i; p < j; p += k - 1) {dp[i][j][m] = min(dp[i][j][m], dp[i][p][1] + dp[p + 1][j][m - 1]);}}dp[i][j][1] = dp[i][j][k] + S[j] - S[i - 1];}}cout << dp[1][n][1] << endl;return 0;
}