【Naptime G】环形dp模板题

原创已于 2022-06-18 09:42:21 修改 · 406 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#动态规划 #算法 #c++

于 2022-06-18 08:36:18 首次发布

c++ 专栏收录该内容

28 篇文章

订阅专栏

这篇博客讲述了如何解决一道关于奶牛贝茜在一天内选择BBB段摆烂时间以最大化效用值的问题，通过动态规划和状态转移方程来处理跨天摆烂限制。作者提供了两种解决方案，一种考虑了每天最早1点开摆的情况，另一种强制在特定时间点摆烂，以确保每个时间段的效用都能计算在内。

[USACO05JAN]Naptime G

题目描述

贝茜是一只非常颓废的奶牛．她的一天被平均分割成 $N$ 段（ $\leq N \leq 3830$ ），但是她要用其中的 $B$ 段时间（ $\leq B \lt N$ ）摆烂。每段时间都有一个效用值 $U_i$ （ $\leq U_i \leq 2 \times 10^5$ ），只有当她摆烂的时候，才会发挥效用。

有了闹钟的帮助，贝茜可以选择任意的时间开摆，当然，她只能在时间划分的边界处开摆、开卷。

贝茜想使所有摆烂效用的总和最大。不幸的是，每一段摆烂的第一个时间阶段都是“开摆”阶段，而旦不记入效用值。

时间阶段是不断循环的圆（一天一天是循环的嘛），假如贝茜在时间 $N$ 和时间 $1$ 摆烂，那么她将得到时间 $1$ 的效用值。

输入格式

第一行两个整数 $N, B$ 。

接下来 $N$ 行，每行一个整数，表示第 $i$ 个时段的效用值。

输出格式

输出最大效用值。

样例 #1

样例输入 #1

样例输出 #1

提示

从第 $4$ 个时段开摆，到第 $1$ 个时段结束开卷。

思路

考虑一个简化的问题，不考虑摆烂跨越两天的情况，即：
状态f[i,j,k]：前 $i$ 个小时中一共睡了 $j$ 个小时，且第 $i$ 个小时的摆烂状态为 $k$ （1睡，0不睡）。
状态转移：

$F [i, j, 0] = M A X (F [i - 1, j, 0], F [i - 1, j, 1])$
$F[i,j,1]=MAX(F[i-1,j-1,0],F[i-1,j-1,1]+U_i)$

很好理解，但这种情况下有一种问题：每天最早1点开摆，此时的效用值无法计入总数。

那再思考另一种情况，强制它前一天第 $N$ 个小时和当天第 $1$ 个小时摆烂，那么第 $1$ 个小时的效用值就可以计算上了。

把分成的两部分分别dp最后取 $M A X$ 即为答案。

CODE：

#include<iostream>
#include<cstring>
using namespace std;
int n,b;
int val[3850];
int dp[3850][3850][2],ans;

int main(){
	cin>>n>>b;
	for(int i=1;i<=n;i++)cin>>val[i];
	
	memset(dp,-0x3f,sizeof dp);
	dp[1][1][1]=dp[1][0][0]=0;
	
	for(int i=2;i<=n;i++){
		dp[i][0][0]=dp[i-1][0][0];
		for(int j=1;j<=b;j++){
			dp[i][j][0]=max(dp[i-1][j][0],dp[i-1][j][1]);
			dp[i][j][1]=max(dp[i-1][j-1][0],dp[i-1][j-1][1]+val[i]);
		}
	}
	ans=max(dp[n][b][0],dp[n][b][1]);
	
	memset(dp,-0x3f,sizeof dp);
	dp[1][1][1]=val[1];//第1个小时也记上了！
	dp[1][0][0]=0;
	
	for(int i=2;i<=n;i++){
		dp[i][0][0]=dp[i-1][0][0];
		for(int j=1;j<=b;j++){
			dp[i][j][0]=max(dp[i-1][j][0],dp[i-1][j][1]);
			dp[i][j][1]=max(dp[i-1][j-1][0],dp[i-1][j-1][1]+val[i]);
		}
	}
	ans=max(ans,dp[n][b][1]);//第N个小时必须得摆！
	
	cout<<ans;
	
	return 0;
}

在以上的基础上再进行滚动数组的优化，使空间复杂的 $n^2→n$ 。

#include<iostream>
#include<cstring>
using namespace std;
int n, b;
int val[3850];
int dp[2][3850][2], ans;

int main() {
	cin >> n >> b;
	for (int i = 1; i <= n; i++)cin >> val[i];

	memset(dp, -0x3f, sizeof dp);
	dp[1][1][1] = dp[1][0][0] = 0;
	
	for (int i = 2; i <= n; i++) {
		for (int j = 0; j <= b; j++) {
			dp[i & 1][j][0] = max(dp[(i - 1) & 1][j][0], dp[(i - 1) & 1][j][1]);
			if(j >= 1)dp[i & 1][j][1] = max(dp[(i - 1) & 1][j - 1][0], dp[(i - 1) & 1][j - 1][1] + val[i]);
		}
	}
	ans = dp[n & 1][b][0];

	memset(dp, -0x3f, sizeof dp);
	dp[1][1][1] = val[1]; //第1个小时也记上了！
	dp[1][0][0] = 0;
	
	for (int i = 2; i <= n; i++) {
		for (int j = 0; j <= b; j++) {
			dp[i & 1][j][0] = max(dp[(i - 1) & 1][j][0], dp[(i - 1) & 1][j][1]);
			if(j >= 1)dp[i & 1][j][1] = max(dp[(i - 1) & 1][j - 1][0], dp[(i - 1) & 1][j - 1][1] + val[i]);
		}
	}
	ans = max(ans, dp[n & 1][b][1]); //第N个小时必须得摆！

	cout << ans;

	return 0;
}