B4172 学习计划 题解

B4172 学习计划 题解

思路

可以将收益式子换一下,设 c i c_i ci 为 a i a_i ai 被分到的段的编号,那收益式子变成 ∑ i = 1 n a i × b c i \sum_{i=1}^n a_i \times b_{c_i} ∑i=1nai×bci。

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

那现在有两种选择。

  1. a i − 1 a_{i-1} ai−1 选的是第 j j j 段,这样的收益是 f i − 1 , j + a i × b j f_{i-1,j}+a_i\times b_j fi−1,j+ai×bj。
  2. a i − 1 a_{i-1} ai−1 选的是第 j − 1 j-1 j−1 段,这样的收益是 f i − 1 , j − 1 + a i × b j f_{i-1,j-1}+a_i\times b_j fi−1,j−1+ai×bj。

整理一下就能得到转移方程 f i , j = max ⁡ ( f i − 1 , j , f i − 1 , j − 1 ) + a i × b j f_{i,j}= \max (f_{i-1,j},f_{i-1,j-1})+a_i\times b_j fi,j=max(fi−1,j,fi−1,j−1)+ai×bj。

由于有负数,初始值要设为 − inf ⁡ -\inf −inf, f 0 , 0 f_{0,0} f0,0 设为 0 0 0。

代码

cpp 复制代码
#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;
}
相关推荐
2202_756749695 分钟前
04 基于sklearn的机械学习-梯度下降(上)
人工智能·算法·机器学习
草莓爱芒果5 分钟前
Spring Boot中使用Bouncy Castle实现SM2国密算法(与前端JS加密交互)
java·spring boot·算法
晚云与城16 分钟前
【数据结构】-----排序的艺术画卷
数据结构·算法·排序算法
weixin_3077791331 分钟前
设计Mock CUDA库的流程与实现
c++·算法·gpu算力
j_xxx404_1 小时前
数据结构:算法复杂度与空间复杂度
c语言·数据结构·算法
dlraba8021 小时前
基于 OpenCV 与 sklearn 的数字识别:KNN 算法实践
opencv·算法·sklearn
yzzzzzzzzzzzzzzzzz1 小时前
leetcode热题——全排列
算法·回溯·全排列
王柏龙1 小时前
mongodb中的哈希索引详解
算法·mongodb·哈希算法
NAGNIP1 小时前
GPT1:通用语言理解模型的开端
后端·算法
NAGNIP2 小时前
GPT-2:让语言模型一统多任务学习江湖
算法