B4172 学习计划 题解

B4172 学习计划 题解

思路

可以将收益式子换一下，设 \(c_i\) 为 \(a_i\) 被分到的段的编号，那收益式子变成 \(\sum_{i=1}^n a_i \times b_{c_i}\)。

很显然的 dp, 设 \(f_{i,j}\) 为将 \(a\) 的前 \(i\) 个数分成 \(j\) 段的最大收益。

那现在有两种选择。

1. \(a_{i-1}\) 选的是第 \(j\) 段，这样的收益是 \(f_{i-1,j}+a_i\times b_j\)。
2. \(a_{i-1}\) 选的是第 \(j-1\) 段，这样的收益是 \(f_{i-1,j-1}+a_i\times b_j\)。

整理一下就能得到转移方程 \(f_{i,j}= \max (f_{i-1,j},f_{i-1,j-1})+a_i\times b_j\)。

由于有负数，初始值要设为 \(-\inf\)，\(f_{0,0}\) 设为 \(0\)。

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

const int N = 2010;
int a[N], b[N];
int f[N][N];
void run() {
	memset(f, -0x3f, sizeof(f));
	int n, m;
	cin >> n >> m;
	for (int i = 1; i <= n; i++) cin >> a[i];
	for (int i = 1; i <= m; i++) cin >> b[i];
	f[0][0] = 0;
	for (int i = 1; i <= n; i++)
		for (int j = 1; j <= min(i, m); j++)
			f[i][j] = max(f[i - 1][j - 1], f[i - 1][j]) + a[i] * b[j];
	cout << f[n][m] << endl;
}
int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	int _ = 1;
	cin >> _;
	while (_--) run();
	return 0;
}