Codeforces Round #638 (Div. 2) E. Phoenix and Berries(DP)(难)

题目链接
Codeforces Round #638 (Div. 2) E. Phoenix and Berries(DP)(难)_第1张图片
Codeforces Round #638 (Div. 2) E. Phoenix and Berries(DP)(难)_第2张图片

题意: 给出n颗树,还有n颗树上红果和蓝果的数量,每个篮放满需要k个果子,放在同一个篮子的果子必须是同种颜色或者是同一棵树上的果子,问最多能放满几个篮子。
思路:dp【i】【j】代表前i棵树还剩j个红果地篮子数,转移地方式有两种,第一种就是选择同一棵树的果子放进篮子里,第二种就是选择同颜色的果子,感觉还剩挺难想的,具体一些细节还剩没怎么明白,留坑以后再回来看看。

#include
using namespace std;
typedef long long ll;
const int maxn=505;
ll dp[maxn][maxn],a[maxn],b[maxn],sum,ans=-1;//dp[i][j]代表前i棵树装完还剩j个红果的最大篮子树 
int main()
{
	int n,k;
	scanf("%d%d",&n,&k);
	for(int i=0;i<n;++i) scanf("%lld %lld",&a[i],&b[i]);
	memset(dp,-1,sizeof(dp));
	dp[0][0]=0;
	for(int i=0;i<n;++i)
	{
		for(int j=0;j<k;++j)
		if(dp[i][j]>=0)
		{
			int t1=sum-dp[i][j]*k-j;//未装篮的蓝果数 
			for(int s=1;s<=a[i]&&s<k;++s)//枚举不同颜色的那个篮子的组成,设有s个红果
			{
				if(b[i]+s>=k)
				{
					int t2=t1+b[i]-(k-s);
					int t=j+a[i]-s;
					dp[i+1][t%k]=max(dp[i+1][t%k],dp[i][j]+t2/k+t/k+1);
				 } 
			} 
			dp[i+1][(j+a[i])%k]=max(dp[i+1][(j+a[i])%k],dp[i][j]+(j+a[i])/k+(t1+b[i])/k);
		 } 
		 sum+=a[i]+b[i];
	}
	for(int i=0;i<=n;++i)
	for(int j=0;j<=k;++j)
	ans=max(ans,dp[i][j]);
	printf("%lld\n",ans);
 } 

你可能感兴趣的:(背包DP)