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;
}