AtCoder - ABC 267 - C D（前缀和 | 01背包）

最新推荐文章于 2023-06-22 12:19:42 发布

奶聂小团子

最新推荐文章于 2023-06-22 12:19:42 发布

阅读量415

点赞数

分类专栏： AtCoder

本文链接：https://blog.csdn.net/m0_64172681/article/details/126697358

版权

动态规划前缀和子序列最大乘积数组

关键词由CSDN通过智能技术生成

AtCoder 专栏收录该内容

33 篇文章 0 订阅

订阅专栏

C - Index × A(Continuous ver.)

题意：

给定一个数组 a,求 a 的一个长度为 m 的连续子序列 b,使 $\sum _{i=1}^{m}i*b_{i}$ 最大,输出最大值。

数据范围：

1 ≤ M ≤ N ≤ 2000

−2× $10^{5}$ ≤ Ai ≤ 2× $10^{5}$

思路：

暴力模拟会超时。需要找等价公式：

在a数组中找长度为 m 的连续子序列 b，使得 $\sum _{i=1}^{m}i*b_{i}$ 最大。因为是连续的，所以a数组中所找序列的每一项的下标，与把序列放在b数组中每一项所对应的下标差值一定是一样的；所以我们可以先算出在a数组中的序列的 $\sum _{i=1}^{m}i*a_{i}$ ，再根据序列放在a，b数组答案的差值计算。

预处理a数组的前缀和，记为 s；a数组的 i * ai 的前缀和记为 b；则当 b 序列以a数组中的第 i 个数开头时，所选序列为 ai, ai+1,……，ai+m-1。在a数组中的值为 s[i+m-1]-s[i-1]，由于a数组的第 i 个数在b数组中下标为 1，所以序列每一项的下标差值为 i -1 ，也就是所选的序列的 m 个数都多算了 i - 1 次才由b数组到a数组，差值为这 m 个数的和的 n - 1 倍，即 (n - 1) * (b[i+m-1] - b[i-1])。

所以有a数组所选的以第 i 个数开头时，在b数组中的值为：s[i+m-1] - s[i-1] - (i-1) * (b[i+m-1]-b[i-1]).

Code:

#include<iostream>
#include<algorithm>
using namespace std;

const int N = 200010;

#define int long long

int m, n;
int a[N], b[N];

void solve()
{
	cin >> n >> m;
	for (int i = 1; i <= n; i++)
	{
		cin >> a[i];
		b[i] = b[i - 1] + i * a[i];    //预处理ai*i的前缀和
		a[i] += a[i - 1];              //预处理a数组的前缀和
	}

	int res = -1e18;
	for (int i = m; i <= n; i++)       //这里枚举的是b序列的最后一个数在a数组的下标i
		res = max(res, b[i] - b[i - m] - (a[i] - a[i - m]) * (i - m));      //则b序列以a数组的第i-m+1个数开头

	cout << res << endl;
}

signed main()
{
	int t = 1;
	//cin >> t;
	while (t--)
	{
		solve();
	}

	return 0;
}

D - Index × A(Not Continuous ver.)

题意：

与C题意相同，但找的 b 是 a数组的子序列，使 $\sum _{i=1}^{m}i*b_{i}$ 最大,输出最大值。

数据范围：

1 ≤ M ≤ N ≤ 2000

−2× $10^{5}$ ≤ Ai ≤ 2× $10^{5}$

思路：

考虑dp，把 n 个数当做有 n 个物品，考虑物品选/不选，m 当做背包容量。相当于01背包问题：

状态表示：f[i,j]表示从前 i 个物品选数量恰好为 j 的所有方案。属性：价值Max

集合划分：以第 i 个物品是否选为依据进行划分，物品的价值 w[i] = a[i]：
1.选第 i 个物品：等价于在前 i-1 个物品中选 j-1 个物品价值，正好第 i 个物品就是所选的第 j 个物品，在b数组中下标为 j，即：f[i,j] = f[i-1,j-1] + j*w[i]
2.不选第 i 个物品：等价于在前 i-1 个物品中选 j 个物品的价值，即：f[i,j] = f[i-1,j]
其实就是01背包问题，转换为一维：f[j] = max(f[j], f[j-1] + j * a[i])。注意枚举 j 时逆序。

Code:

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;

const int N = 200010;

#define int long long

int m, n;
int a[N];
int f[N];

void solve()
{
	cin >> n >> m;
	for (int i = 1; i <= n; i++)cin >> a[i];

    //01背包问题恰好装满的初始化
	memset(f, -0X3f, sizeof(f));
	f[0] = 0;

	for (int i = 1; i <= n; i++)
		for (int j = m; j >= 1; j--)
			f[j] = max(f[j], f[j - 1] + j * a[i]);

	cout << f[m] << endl;
}

signed main()
{
	int t = 1;
	//cin >> t;
	while (t--)
	{
		solve();
	}

	return 0;
}

------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
吐槽：自我感觉D比C容易，好吧，其实是我菜当时写时乍一看写不出来脑子就不转了>-<。题目注意点：

1.C题让我感受到前缀和还可以这样用，不能单单局限于单纯的数的和。

2.D题预处理时注意数据范围。还有C题的也是res的初始化。