【01背包 && bitset优化 && 取 10 个数构成 78 的方案数】HDU - 5890 Eighty seven

Step1 Problem

给你 n 个数，Q 次询问，每次询问删除三个下标对应的数，在剩下的数的基础上选择不超过 10 个数能否组成 87.
数据范围：
N <= 50, Q <= 1e5.

Step2 Ideas:

ans[i][j][k]：代表删除下标 i, j, k 后，在剩下的数的基础上选择不超过 10 个能否组成 87.
我们预处理 ans[i][j][k]:
正常 01 背包时间复杂度：O(n*10*87)
我们有 bitset 可以把 87 优化掉：O(n*10)
总的时间复杂度 O(n^4*10);
bitset<87> dp[15];
dp[i][j]：代表用了 i 个数，是否构成了 j .
dp[j] |= dp[j-1] << a[i]; //相当于 dp[j-1] 所有的值都加 a[i]
dp[10][87] 为真，代表能构成，反之不能构成。

Step3 Code:

#include<bits/stdc++.h>
using namespace std;
const int N = 55;
int a[N], n;
bool ans[N][N][N];
bitset<88> dp[15];
bool ok(int x, int y, int z)
{
    for(int i = 1; i <= 10; i++) dp[i].reset();
    dp[0][0] = 1;
    for(int i = 1; i <= n; i++)
    {
        if(i == x || i == y || i == z || a[i] > 87) continue;
        for(int j = 10; j >= 1; j--)
        {
            dp[j] = dp[j] | (dp[j-1]<<a[i]);
        }
    }
    return dp[10][87];
}
int main()
{
  //  ios::sync_with_stdio(false);
    int T;
    scanf("%d", &T);
    while(T--)
    {
        scanf("%d", &n);
        for(int i = 1; i <= n; i++)
            scanf("%d", a+i);
        for(int i = 1; i <= n; i++)
        {
            for(int j = i; j <= n; j++)
            {
                for(int k = j; k <= n; k++)
                {
                    int flag;
                    if(ok(i, j, k)) {
                        flag = 1;
                    }
                    else flag = 0;
                    ans[i][j][k] = ans[i][k][j] = flag;
                    ans[j][i][k] = ans[j][k][i] = flag;
                    ans[k][i][j] = ans[k][j][i] = flag;
                }
            }
        }
        int m;
        scanf("%d", &m);
        int q[5];
        while(m--)
        {
            scanf("%d %d %d", q, q+1, q+2);
            if(ans[q[0]][q[1]][q[2]]) printf("Yes\n");
            else printf("No\n");
        }
    }
    return 0;
}