题意,t组样例,每个样例先给你一个n,下面是n个数分别表示第几个秤砣的重量,再给你一个数m,下面是m个数,表示你要判断是否能称出的物品重量,秤砣可以放两边,如果可以称出来输出YES,否则输出NO
我首先简单说一下二进制枚举,对于 4 7 12 18 四个重量,我们可以用四位数字表示例如0100,0101,如果第i位数字是1表示用第i个秤砣(从右往左看),为零则不用。那么0100表示只用第3个秤砣,0101表示用第1,3个秤砣.我们要枚举每种状态,那么我们怎么枚举呢?我们以上面的例子为例,n为4,1可以看成0001,24-1可以看成1111,1~24-1之间的数字分别对应一种状态,n个秤砣就可以用1-2n-1表示每一种状态,那我们怎么判断每位上的数字呢?1&1=1,1&0=0,所以我们只要用在每位用1和他做&运算即可.
1<<0(1) 判断第1位数字
1<<1(10) 判断第2位数字
1<<2(100)判断第3位数字
…直到第n位
基本代码实现
for(int i=0;i<(1<<n);i++)//枚举每种情况
for(int j=0;j<n;j++)//判断每一位上的数字
if(i&(1<<j))
思路:回归正题,这道题的话我们可以用二进制枚举来记录所有可以称的重量,
但是:秤砣可以放两边,我们可以想到把挑出来的秤砣放到物品那边几个,我们可以用
for(int k=0;k<n;k++) if(s-a[k]>0) mp[s-a[k]]=1;
实现,为什么只减去一个秤砣的重量呢,对于我们挑出的k个秤砣,其中kk个放在物品边,那么之前一定有过挑出(k-kk+1)个秤砣,挑出1个放在左边的操作,所以我们可以只考虑当前的状态挑出一个放在物品侧的情况,我们每求得一个s,就放在物品那侧一个就是所有情况
AC code
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<map>
#include<queue>
#include<stack>
using namespace std;
int a[30000],mp[50000];
void solve()
{
memset(mp,0,sizeof(mp));
int flag=0,n;
scanf("%d",&n);
for(int i=0;i<n;i++)
scanf("%d",&a[i]);
int m;
scanf("%d",&m);
for(int i=0;i<(1<<n);i++)
{ int s=0;
for(int j=0;j<n;j++)
if(i&(1<<j))
{
s+=a[j];
}
mp[s]=1;
for(int k=0;k<n;k++)//每次从挑出的秤砣中选一个放在物品侧
if(s-a[k]>0) mp[s-a[k]]=1;
}
for(int i=0;i<m;i++)
{
int x;
scanf("%d",&x);
if(mp[x])printf("YES\n");
else printf("NO\n");
}
}
int main()
{
int t;
scanf("%d",&t);
while(t--)
solve();
}