Comet OJ - Contest #11 B.usiness DP

问题连接

问题描述

问题分析

思路：dp[i][j]表示第i天且有j个结点第n天所能获得的最大结点数。根据题意dp[1][0]=0。然后从第2天起(因为第一天结点数为0，无操作)找到上一天的传递给这一天的解，从中获取最大值。后进行m种操作(完全背包)可以找到操作完后的各个状态的最优解。如此进行完n天后，因为第n天结束后每个状态还会有对应收入，所以再从第n天的所有状态中找到最大值即是答案。
注:Max=2000，是已有结点数可能的最大值，因为w[i]不大于1000，k也不大于1000，而且i>k时为0，所以最大情况是在i=1000的时候w[i]=1000,所以个数最大为2000。而dp数组的第二维大小设置为3005是因为避免dp[i][Max+f(Max)]中Max+f(Max)超过数组表示范围。

其它心得：
①这题的完全背包。以前见到的完全背包的形式是内循环是顺序的，而这题的内循环是逆序的，所以在看他人的题解的时候一直心存质疑。后想明白了题目的操作过程，这样做是因为不是像之前的从数值小的状态传递给数值大的状态，而是从数值大的状态传递给数值小的状态。
②像01背包的二维数组可以改成一维数组优化空间，关键是在遍历的时候避免一些状态交错使用。

代码如下

#include<bits/stdc++.h>
using namespace std;
const int N=105;
const int M=1e3+5;
const int Max=2e3;
#define endl '\n'

int n,m,c;
int a[N],b[N];
int w[M];

int dp[N][3005];

int f(int x){
	if(x>c) return 0;
	return w[x];
}

int main(){
	ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
	cin>>n>>m>>c;
	for(int i=0;i<=c;i++) cin>>w[i];
	for(int i=0;i<m;i++) cin>>a[i]>>b[i];
	
	memset(dp,-1,sizeof(dp));//初始化无效标记
	dp[1][0]=0;
	for(int i=2;i<=n;i++){
		for(int j=0;j<=Max;j++){//从第i-1天获取第i天未操作时的最大结点数
			if(dp[i-1][j]==-1) continue;//无效状态跳过，下同
			dp[i][j+f(j)]=max(dp[i][j+f(j)],dp[i-1][j]+f(j));
			//dp[i][j+f(j)]可以选择不变动，也可以是从上一天的dp[i-1][j]加了f(j)过来的。
		}
		for(int k=0;k<m;k++){//进行完全背包,分别找到对应状态拥有的最大结点数
			for(int j=Max;j>=a[k];j--){
				if(dp[i][j]==-1) continue;
				dp[i][j-a[k]]=max(dp[i][j-a[k]],dp[i][j]-a[k]+b[k]);
				/*dp[i][j-a[k]]的状态可以从dp[i][j]获得，
				dp[i][j]失去a[k]个结点，未来会再获得b[k]个结点，
				所以记总数的时候是dp[i][j]-a[k]+b[k]。
				也可能已经从其它状态获取的最优解而不用更新。*/
			}
		}
	}
	int ans=0;
	for(int i=0;i<=Max;i++)//第n天操作完后，还会获得f(x)的顶点，从中找最大值
	if(dp[n][i]!=-1) ans=max(ans,dp[n][i]+f(i));
	cout<<ans<<endl;
	return 0;
}

内存优化后(只是简单的修改了一下)

#include<bits/stdc++.h>
using namespace std;
const int N=105;
const int M=1e3+5;
const int Max=2e3;
#define endl '\n'

int n,m,c;
int a[N],b[N];
int w[M];

int dp[3005];//二维数组变为一维数组

int f(int x){
	if(x>c) return 0;
	return w[x];
}

int main(){
	ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
	cin>>n>>m>>c;
	for(int i=0;i<=c;i++) cin>>w[i];
	for(int i=0;i<m;i++) cin>>a[i]>>b[i];
	
	memset(dp,-1,sizeof(dp));
	dp[0]=0;
	for(int i=2;i<=n;i++){
		for(int j=Max;j>=0;j--){
		//避免这一天的状态再被这天的其它状态利用，所以从大到小遍历
			if(dp[j]==-1) continue;
			dp[j+f(j)]=max(dp[j+f(j)],dp[j]+f(j));
		}
		for(int k=0;k<m;k++){
			for(int j=Max;j>=a[k];j--){
				if(dp[j]==-1) continue;
				dp[j-a[k]]=max(dp[j-a[k]],dp[j]-a[k]+b[k]);
				//这部分与天数无关，直接去天数即可
			}
		}
	}
	int ans=0;
	for(int i=0;i<=Max;i++)
	if(dp[i]!=-1) ans=max(ans,dp[i]+f(i));
	cout<<ans<<endl;
	return 0;
}