牛客 小M和天平

题目:https://ac.nowcoder.com/acm/problem/13586
这题很明显是一个背包(做多了可能就有感觉了吧),设f[i][j]为选了前i个物品能否凑出体积为j的物品,然后用01背包乱转移就可以了。但这题要注意的是不能用一维的背包写法,必须用二维,因为这个背包会同时往前和往后更新状态(毕竟石子相加和相减都能凑出物品),用一维的话会导致变成完全背包(不懂可以手动模拟?),然后就可以写代码了

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;
int a[105],f[105][10305],n,m;
int main()
{
	while(scanf("%d",&n)!=EOF)
	{
		f[0][0]=1;
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		for(int j=0;j<=10000;j++)//背包,若j能被凑出,那么j+a[i]和j-a[i]也必定能凑出,因为石子可以放天平两边 
		{
			f[i][j]=f[i-1][j];
		}
		for(int j=0;j<=10000;j++)
		{
			f[i][j]|=f[i-1][j+a[i]];
			f[i][j]|=f[i-1][abs(j-a[i])];
			f[i][j]|=f[i-1][j];
		}
	}
	cin>>m;
	for(int i=1;i<=m;i++)
	{
		int k;
		scanf("%d",&k);
		if(k>10000)//这个是数据范围决定的 
		{
			printf("NO\n");
			continue;
		}
		if(f[n][k])
		{
			printf("YES\n");
		}
		else
		{
			printf("NO\n");
		}
	}
	}
 } 

猜你喜欢

转载自blog.csdn.net/qq_37073764/article/details/103454762