【二进制枚举】Jam's balance HDU - 5616

题目传送门

题目 传送门

题意

有一个天平，和n个秤砣，m次访问，每次给出一个重量k，能称出重量则输出YES，否则输出NO

思路

二进制枚举出n个秤砣所有能表示出的重量，用vis标记，即可做到O（1）的访问

当我们有表示w[3 ]= {1、5、7}的3个秤砣时我们可以称量出的质量有：
1、5、7
6、 8、12、13（1+5、1+7、5+7、1+5+7）
到这里我们就可以用二进制枚举列出所有情况了

你以为完了吗？ NONONONO！！！

我们还可以称量出
4、 6、2（5 - 1、7 - 1、7 - 5）三种质量
所以枚举完后还要需要一个循环减去w[i]。

代码如下

#include <cstdio>
#include <iostream>
using namespace std;
const int N = 1e7;
bool vis[N];
int w[N];
int main()
{
    int t;
    cin >>t;
    while (t--) {
        memset(vis,0,sizeof(vis));
        int n,m,k;
        cin >> n;
        for (int i = 0; i < n; ++i)
            cin >> w[i];
        for (int i = 0; i < (1 << n); ++i) {//二进制枚举
            int sum = 0;
            for (int j = 0; j < n; ++j) {
                if(i & (1 << j))
                    sum += w[j];
            }
            vis[sum] = 1;
            for (int j = 0; j < n; ++j) {//5-1、7-1、7-5的情况
                if(sum - w[j] > 0)
                    vis[sum-w[j]] = 1;
            }
        }
        cin >> m;
        for(int i = 0; i < m; ++i){
            cin >> k;
            if(vis[k]) puts("YES");
            else puts("NO");
        }
    }
    return 0;
}