把四堆石子合成一堆
四堆石子 4、1、2、3,每次只能合并相邻两堆,代价是这两堆之和。求把它们合成一堆的最小总代价。运行下面这段程序:
#include <algorithm>
#include <iostream>
#include <string>
#include <vector>
using namespace std;
using Item = pair<int, int>; // (重量, 价值)
const vector<Item> ITEMS = {{3, 8}, {4, 9}, {5, 11}};
long long merge_cost(const vector<int>& a) {
int n = a.size();
const long long INF = 1000000000000000000LL;
vector<long long> pre(n + 1, 0);
for (int i = 0; i < n; i++) pre[i + 1] = pre[i] + a[i];
vector<vector<long long>> dp(n, vector<long long>(n, 0));
for (int L = 2; L <= n; L++) {
for (int i = 0; i + L - 1 < n; i++) {
int j = i + L - 1;
dp[i][j] = INF;
for (int k = i; k < j; k++) {
long long t = dp[i][k] + dp[k + 1][j] + pre[j + 1] - pre[i];
if (t < dp[i][j]) dp[i][j] = t;
}
}
}
return dp[0][n - 1];
}
int main() {
cout << merge_cost({4, 1, 2, 3}) << endl;
}
(本题用 g++ -std=c++17 -O0 编译。)
全部评论