other/knuth.cpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15

ll calc(int n, int k, const vector<vector<ll>> &C) {
	vector<vector<ll>> dp(k, vector<ll>(n, inf));
	vector<vector<int>> opt(k, vector<int>(n + 1, n - 1));

	for (int i = 0; i < n; i++) dp[0][i] = C[0][i];
	for (int i = 1; i < k; i++) {
		for (int j = n - 1; j >= 0; --j) {
			opt[i][j] = i == 1 ? 0 : opt[i - 1][j];
			for (int k = opt[i][j]; k <= min(opt[i][j + 1], j - 1); ++k) {
				if (dp[i][j] <= dp[i - 1][k] + C[k + 1][j]) continue;
				dp[i][j] = dp[i - 1][k] + C[k + 1][j];
				opt[i][j] = k;
	}}}
	return dp[k - 1][n - 1];
}