优化这段代码:#include <iostream> #include <cstring> using namespace std; const int MAXN = 1005; int w[MAXN]; // 存放每堆石子的数量 int sum[MAXN];// 存放前缀和 int dp[MAXN][MAXN];// dp数组 int main() { int n; cin >> n; for (int i = 1; i <= n; i++) { cin >> w[i]; sum[i] = sum[i - 1] + w[i]; // 计算前缀和 } for (int len = 2; len <= n; len++) // 枚举子区间长度 { for (int i = 1; i + len - 1 <= n; i++) // 枚举子区间起点 { int j = i + len - 1; // 子区间终点 dp[i][j] = INT_MAX; // 初始化为最大值 for (int k = i; k < j; k++) // 枚举区间分割点 { dp[i][j] = min(dp[i][j], dp[i][k] + dp[k + 1][j] + sum[j] - sum[i - 1]); } } } cout << dp[1][n] << endl; // 输出最小花费 return 0; } 输入格式为第一行第一行两个正整数n(n≤100000)和k,表示有n堆石子,每次至少选2堆、至多选k堆石子合并。第二行有n个数,分别表示每堆石子的个数
时间: 2024-01-21 22:04:54 浏览: 74
这段代码是求解将n堆石子合并成一堆的最小代价,可以使用区间DP进行优化。
优化后的代码如下所示:
```c++
#include <iostream>
#include <cstring>
using namespace std;
const int MAXN = 1005;
int w[MAXN]; // 存放每堆石子的数量
int sum[MAXN];// 存放前缀和
int dp[MAXN][MAXN];// dp数组
int main() {
int n, k;
cin >> n >> k;
for (int i = 1; i <= n; i++) {
cin >> w[i];
sum[i] = sum[i - 1] + w[i]; // 计算前缀和
}
memset(dp, 0x3f, sizeof(dp)); // 初始化为最大值
for (int i = 1; i <= n; i++) {
dp[i][i] = 0; // 单个区间的代价为0
}
for (int len = 2; len <= n; len++) { // 枚举子区间长度
for (int i = 1; i + len - 1 <= n; i++) { // 枚举子区间起点
int j = i + len - 1; // 子区间终点
for (int s = i; s < j; s++) { // 枚举区间分割点
for (int t = 2; t <= k && s + t - 1 <= j; t++) { // 枚举合并堆数
dp[i][j] = min(dp[i][j], dp[i][s] + dp[s + 1][s + t - 1] + dp[s + t][j] + sum[j] - sum[i - 1]);
}
}
}
}
cout << dp[1][n] << endl; // 输出最小花费
return 0;
}
```
主要的优化点如下:
1. 添加了一个变量k,表示每次至多选k堆石子合并。
2. 初始化dp数组时,只有单个区间的代价为0,其他值均为最大值。
3. 在枚举区间分割点时,增加了一个循环,枚举合并的堆数。
阅读全文