box(盖子左移或不动或右移）

__night_

于 2023-11-07 17:51:52 发布

阅读量71

点赞数 1

CC 4.0 BY-SA版权

分类专栏： 2023牛客多校训练营文章标签：算法

本文链接：https://blog.youkuaiyun.com/way_back__/article/details/134272702

2023牛客多校训练营专栏收录该内容

26 篇文章

订阅专栏

该文章介绍了如何使用C++编程语言实现一个动态规划算法来解决盒子移动问题，计算在给定限制下移动盒子以获取最大价值的问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目

#include <bits/stdc++.h>
using namespace std;
const int  N = 1000000;
long long f[N + 5][3];//f[i][j]表示前i个盒子，第i个盒子移动j-1（即0为左移，1不动，2右移）的最大价值（若第i个右移，答案也加上a[i + 1])
long long a[N + 5];
int pos[N + 5], cnt, b[N + 5];
int main()
{
	int  n;
	scanf("%d", &n);
	for (int  i = 1; i <= n; i++)
		scanf("%lld", &a[i]);
	for (int i = 1; i <= n; i++)
	{
		scanf("%d", &b[i]);
		if (b[i])
			pos[++cnt] = i;
	}
	if (!cnt)
	{
		printf("0");
		return 0;
	}
	memset(f, 0xcf, sizeof(f)); // 初始化为全-inf
	f[1][1] = a[pos[1]];
	if (pos[1] > 1)
		f[1][0] = a[pos[1] - 1];
	if (pos[1] < n)
		f[1][2] = a[pos[1] + 1];
	for (int i = 2; i <= cnt; i++)
		for (int j = 0; j <= 2; j++)
			for (int k = 0; k <= 2; k++)
				if (pos[i - 1] + k - 1 < pos[i] + j - 1 && pos[i] + j - 1 <= n)
					// 不越界，盖子之间相对位置关系保持（即不能重叠，也不会交换位置（没意义））
					f[i][j] = max(f[i][j], f[i - 1][k] + a[pos[i] + j - 1]);
	printf("%lld", max({f[cnt][0], f[cnt][1], f[cnt][2]}));
	return 0;
}