CF254E Dormitory

Problem

每天有一定量的食物，食物的保质期都是两天即今天和明天，每天有一些朋友，可以用食物投喂他们，求n天最多投喂多少次朋友。

Solution

• 如果食物的保质期一天可以用贪心解决，食物的保质期是两天就要想办法将每一天当天新获得的食物的剩余体现在状态上（这样才能合法地转移）。
• a(i)为第i天所得食物，dp(i,j)表示前i天给i+1天剩了j食物所能投喂的最大次数，dp(i,j)=max(dp(i-1,k)+用a(i)-j+k最多能投喂的数量)，注意对于每个i，j的范围是[0,a(i)]而非[0,a(i)-v]，因为每一天自己吃当天的食物和昨天的食物是不等价的（可以给明天保留更多食物），因此在过程中要考虑涉及状态的合法性，同时时间复杂度要优化到n^3。

Code

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<vector>
#include<cstring>
using namespace std;
const int maxn = 405;
int n,v,a[maxn],dp[maxn][maxn],m,last[maxn][maxn];
vector<int>path[maxn][maxn];
struct node{
	int num,x;
}nd[maxn][maxn];
bool cmp(const node &a,const node &b){
	return a.x<b.x;
}
void dfs(int i,int j)
{
	if(i>1)
	dfs(i-1,last[i][j]);
	cout<<path[i][j].size();
	for(int k=0;k<path[i][j].size();k++)
	cout<<' '<<path[i][j][k];
	cout<<endl;
}
int main()
{
	freopen("input.txt","r",stdin); 
	freopen("output.txt","w",stdout);
	cin>>n>>v;
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	cin>>m;
	memset(dp,-1,sizeof(dp));
	dp[0][0]=0;
	for(int i=1,l,r,f;i<=m;i++)
	{
		scanf("%d%d%d",&l,&r,&f);
		for(int j=l;j<=r;j++)
		nd[j][++nd[j][0].x]={i,f};
	}
	for(int i=1;i<=n;i++)
	{
		sort(nd[i]+1,nd[i]+1+nd[i][0].x,cmp);
		for(int j=2;j<=nd[i][0].x;j++)
			nd[i][j].x+=nd[i][j-1].x;
		nd[i][0].x=0;
		
	}
	for(int i=1;i<=n;i++)
	{
		for(int j=a[i];j>=0;j--)
		{
			int ans=0;
			for(int k=0;k<=a[i-1];k++)
			{
				int ans0=0,ans1=a[i]-j+k-v;
				if(ans1<0||dp[i-1][k]==-1)	continue;	
				if(!ans)
				for(int o=8;o>=0;o--)
				{
					if(nd[i][ans+(1<<o)].x&&ans1>=nd[i][ans+(1<<o)].x)
					ans+=(1<<o);
				}
				if(dp[i-1][k]+ans>dp[i][j])
				{
					last[i][j]=k;
					ans0=0;
					dp[i][j]=dp[i-1][k]+ans;
					path[i][j].clear();
					for(int o=1;o<=ans;o++)
						path[i][j].push_back(nd[i][o].num);
				}
				if(nd[i][ans+1].x)
				{
					k=nd[i][ans+1].x-(ans1-k)-1;
					ans++;
				}
			}
		}
	}
	cout<<dp[n][0]<<endl;
	dfs(n,0);
	return 0;
}