B-智乃买瓜

链接:登录—专业IT笔试面试备考平台_牛客网
来源:牛客网
 

有一人前来买瓜。
“哥们儿,这瓜多少钱一斤呐”
“两块钱一斤”
“What's up,这瓜皮是金子做的,还是瓜粒子是金子做的”

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

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

  1. 购买一整个重量为wiw_iwi​的西瓜
  2. 把瓜劈开,购买半个重量为wi2\frac{w_i}{2}2wi​​的西瓜
  3. 不进行购买操作

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

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

输入描述:

 

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

若N{N}N不为0{0}0,接下来一行N{N}N个正偶数wi(2≤wi≤2×103)w_i (2 \leq w_i \leq2\times 10^3)wi​(2≤wi​≤2×103)表示每个西瓜的重量。

输出描述:

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

输入

3 6
8 2 4

输出

1 2 1 3 2 3

思路:变种的01背包问题。比赛时想到时背包了,但是脑抽了,想着这个瓜劈了半边,还剩半边能再选,就一直卡着不知道怎么去处理。还是审题问题。其实做法就是一个瓜每次判断买一个或者买半个就好了,很简单的模板题,背景挺有意思的(可能就是因为脑子里就是劈瓜的画面,弄得不会做了)

#include<bits/stdc++.h> 
using namespace std;
#define ll long long
ll n,m;
ll w[1008],v[1008];

const int MOD = 1e9+7;

int main() {
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	cin >> n >> m;
	for(int i = 1; i <= n; i++)
		cin >> w[i];
	v[0] = 1;
	for(ll i = 1; i <= n; i++ ) {
		for(ll j = m; j; j--) {
			if( j >= w[i]/2)
				v[j] += v[j - w[i]/2];
			if( j >= w[i])
				v[j] += v[j-w[i]];
			 v[j] %= MOD;
		}
	}
	for(ll i = 1; i <= m; i++) {
		if(i != m )
			cout<<v[i]<<" ";
		else
			cout<<v[i];
	}
		
   	return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值