智乃买瓜+（another version)--动态规划

题目描述

有一人前来买瓜。 “哥们儿,这瓜多少钱一斤呐”

“两块钱一斤”

“What's up，这瓜皮是金子做的,还是瓜粒子是金子做的”

智乃来到水果摊前买瓜，水果摊上贩卖着N个不同的西瓜，第i个西瓜的重量为wi​。智乃对于每个瓜都可以选择买一个整瓜或者把瓜劈开买半个瓜，半个瓜的重量为wi/2。

也就是说对于每个西瓜，智乃都有三种不同的决策：

1.购买一整个重量为wi​的西瓜

2.把瓜劈开，购买半个重量为wi/2的西瓜

3.不进行购买操作

为了简化题目，我们保证所有瓜的重量都是一个正偶数。

现在智乃想要知道，如果他想要购买西瓜的重量和分别为k=1,2,3...M时，有多少种购买西瓜的方案，因为这些数字可能会很大，请输出方案数对10^9+7取余数后的结果。

输入

第一行输入两个整数N,M(0≤N≤103,1≤M≤103),分别表示西瓜的数目N，以及查询的重量上限为M。

若N不为0，接下来一行N个正偶数wi(2≤wi≤2×103)表示每个西瓜的重量。

输出

输出一行M个数字，分别表示购买西瓜的重量和为k=1,2,3...M时，有多少种购买西瓜的方案，因为这些数字可能会很大，请输出方案数对10^9+7取余数后的结果。

样例输入

3 6
8 2 4

样例输出

1 2 1 3 2 3

分析

这是背包问题，M代表背包容量，wi代表物品重量，开一个dp[N][M]数组，dp[i][j]表示现在放到第i个西瓜，且当前的重量为j，方案数为dp[i][j]。

DP转移方程：

dp[0][0]=1

dp[i][j]=dp[i−1][j−wi]+dp[i−1][j−wi/2]+dp[i−1][j]（三种决策）

当j<wi​或者j<wi/2​​时没有dp[i][j]。

处理时将二维数组用滚动数组优化成一维。

代码

#include<bits/stdc++.h>
using namespace std;
const int MAXN=1005;
int n,m,w;
const long long mod=1e9+7;
long long dp[MAXN]; 
int main()
{
	dp[0]=1;
	scanf("%d %d",&n,&m);
	for(int i=1;i<=n;++i)
	{
		scanf("%d",&w);
		for(int i=m;i;--i)
		{
			if(i>=w/2){
			dp[i]%=mod;
			dp[i]+=dp[i-w/2];
			dp[i]%=mod;
		}
			if(i>=w){
				dp[i]%=mod;
			dp[i]+=dp[i-w];
			dp[i]%=mod;} 
		}
	}
	for(int i=1;i<=m;++i)
	{
		printf("%lld%c",dp[i]%mod," \n"[i==m]);
	}
	return 0;
}

(another version)

题目

现在智乃知道，购买西瓜的重量和分别为k=1,2,3...M时，购买西瓜的方案的种类数对10^9+7取余数后的结果。

她想要还原水果摊贩卖的这若干个不同的西瓜重量分别为多少，请你构造一个贩卖N{N}N个不同西瓜的水果摊，其购买西瓜的重量和满足智乃的要求，当出现有多种符合题目描述的答案时，你只用输出任意一种。

我们保证输入的数据至少存在一个N≤103的合法解。

输入

第一行输入一个正整数M(1≤M≤103)，表示智乃所知购买西瓜质量和的上限。

接下来一行M个整数，第i个整数表示购买西瓜的重量和为i时，购买西瓜的方案的种类数对10^9+7取余数后的结果。

输出

首先输出一个整数N，西瓜的数目。要求你给出的N大小范围在[0,10^3]之间

接下来一行N个正偶数wi​，表示每个每个西瓜的重量。要求你给出的wi​大小范围在[2,2×10^3]之间，且wi为偶数。

输入数据保证，至少存在一组在输出范围内的合法解。

样例输入

6
1 2 1 3 2 3

样例输出

3
8 2 4

分析

这是上一道题的逆向输入和输出，逆向思维把转移方程倒过来

买瓜时的转移方程

dp[i][j]=dp[i−1][j−wi​]+dp[i−1][j−wi/2​​]+dp[i−1][j]

则

dp[i][j]−dp[i−1][j−wi]−dp[i−1][j−wi/2]=dp[i−1][j]（移项）

dp[i−1][j]=dp[i][j]−dp[i−1][j−wi]−dp[i−1][j−wi/2]

处理时也用滚动数组优化一下

dp[j]=(dp[j]−dp[j−wi]−dp[j−wi/2])%mod

如果dp[1]>0, 说明有dp[1]个重量为2的瓜, 因为重量和为1的方案只能由1个重量为2的瓜买一半得到

若dp[2]>0, 则说明有dp[2]个重量为2∗2的瓜，因为dp[2]的转移方程中减去了dp[1]的方案数，所以

do[2]只能由1个重量为4的瓜买一半得到

以此类推, 从小到大第一个大于0的dp[i]说明有dp[i]个重量为2∗i的瓜

代码

#include<bits/stdc++.h>
using namespace std;
const long long mod=1e9+7;
const int MAXN=1005;
long long dp[MAXN];
int m;
vector<int>ans;
int main()
{
	dp[0]=1;
	scanf("%d",&m);
	for(int i=1;i<=m;++i)
	{
		scanf("%lld",&dp[i]);
	}
	for(int i=1;i<=m;i++)
	{
		while(dp[i])
		{
			ans.push_back(i*2);
			assert(ans.size()<=1000);
			for(int j=1;j<=m;j++)
			{
			 if (j - i >= 0)
                    dp[j] =( dp[j] - dp[j - i]+ mod) % mod ;
                if (j - 2 * i >= 0)
                    dp[j] = (dp[j] - dp[j - 2 * i]+ mod) % mod ;
			}
		}
	}
	printf("%d\n",ans.size());
	for(int i=0;i<ans.size();++i)
	{
		printf("%d%c",ans[i]," \n"[i+1==ans.size()]);
	}
	return 0;
}