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

#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;
}