盾神与砝码称重 DFS

问题 1548: [蓝桥杯][算法提高VIP]盾神与砝码称重

时间限制: 1Sec 内存限制: 128MB 提交: 504 解决: 183

题目描述

有一天，他在宿舍里无意中发现了一个天平！这  个天平很奇怪，有n个完好的砝码，但是没有游码。盾神为他的发现兴奋不已！于是他准备去称一称自己的东西。他准备好了m种物品去称。神奇的是，盾神一早就  知道这m种物品的重量，他现在是想看看这个天平能不能称出这些物品出来。但是盾神稍微想了1秒钟以后就觉得这个问题太无聊了，于是就丢给了你。

数据规模和约定
1< =n< =24,  1< =m< =10.

输入

第一行为两个数，n和m。 
第二行为n个数，表示这n个砝码的重量。 
第三行为m个数，表示这m个物品的重量。 

输出

输出m行，对于第i行，如果第i个物品能被称出，输出YES否则输出NO。 

样例输入

4  2 
1  2  4  8 
15  16 

样例输出

YES
NO

思路：

假设在左边放物品，右边放砝码，那么左边==右边重量时才能称出来。

那么我们也可以选砝码放到左边。这一点是关键。

那么对于每一个砝码有三个状态：不放、放左边、放右边。

代码：

#include<iostream>
#include<algorithm>
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define dep(i,a,b) for(int i=a;i>=b;i--)
#define ll long long
#include<string.h>
#include<queue>
#include<stack>
#include<cstdio>
using namespace std;
const int maxn=20+6;
#define mod 1000000007
#define INF 0x3f3f3f3f
int dx[]= {-1,1,0,0};
int dy[]= {0,0,-1,1};
int n,m;
int a[maxn];//西交
bool vis[maxn];
int ans;
void dfs(int now,int sum,int x)
{
    if(sum==x)
    {
        ans=1;
        return ;
    }
    if(now>n)
        return;
    dfs(now+1,sum+a[now],x);//放砝码那边
    dfs(now+1,sum,x);
    dfs(now+1,sum-a[now],x);//放物品那边
    return ;
}
int main()
{
    cin>>n>>m;
    int flag=0;
    rep(i,1,n)
    {
        cin>>a[i];
    }
    rep(i,1,m)
    {
        int x;
        cin>>x;
        ans=0;
        dfs(1,0,x);
        if(ans)
            cout<<"YES"<<endl;
        else
            cout<<"NO"<<endl;
        memset(vis,0,sizeof(vis));
    }
    return 0;
}