hdu5890(二维01背包+bieset优化)

链接：点击打开链接

题意：给出n个数和q次询问，每次删除1到3个数，问使否能从剩下的数中选10个数和为87

代码：

#include <bitset>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
int n,a[55],sign[55][55][55];
bitset<90> dp[11];
int judge(int x,int y,int z){
    int i,j;
    for(i=1;i<=n;i++)
    dp[i].reset();
    dp[0][0]=1;
    for(i=1;i<=n;i++){                          //用bitset代替二维数组，可以
        if(i!=x&&i!=y&&i!=z)                    //直接进行容量的转移
        for(j=9;j>=0;j--)
        dp[j+1]|=(dp[j]<<a[i]);
    }
    if(dp[10][87]==1)
    return 1;
    return 0;
}
int main(){
    int t,i,j,k,q,x,y,z;
    scanf("%d",&t);
    while(t--){
        scanf("%d",&n);
        for(i=1;i<=n;i++)
        scanf("%d",&a[i]);
        memset(sign,0,sizeof(sign));
        for(i=1;i<=n;i++){                      //因为n只有50因此可以提前预处理出来
            for(j=i;j<=n;j++){                  //而且很明显是一个二维背包，但是复杂度
                for(k=j;k<=n;k++){              //太高因此不能直接用二维背包求解
                    if(judge(i,j,k)){
                        sign[i][j][k]=sign[i][k][j]=1;
                        sign[j][i][k]=sign[j][k][i]=1;
                        sign[k][i][j]=sign[k][j][i]=1;
                    }
                }
            }
        }
        scanf("%d",&q);
        while(q--){
            scanf("%d%d%d",&x,&y,&z);
            if(sign[x][y][z])
            puts("Yes");
            else
            puts("No");
        }
    }
    return 0;
}