题目大意:给50张以内的牌,每次取走三张以内的牌,问在剩下的牌里抽10张是否能组成87. 询问有10w次
题解:提前打表,用bitset优化背包转移。
bitset的复杂度应该是O(n)/一个常数
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <bitset>
using namespace std;
bitset<90>dp[11];
int ans[60][60][60];
int n,sa[100],m,q[10];
bool check(int x,int y,int z)
{
for(int i=0;i<=10;i++)dp[i].reset();
dp[0][0]=1;
for(int i=1;i<=n;i++)
{
if(i!=x&&i!=y&&i!=z)
for(int t=10;t>=1;t--)
dp[t]|=dp[t-1]<<sa[i];
}
if(dp[10][87]==1)
return 1;
return 0;
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
memset(ans,0,sizeof(ans));
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",&sa[i]);
for(int i=1;i<=n;i++)
for(int j=i;j<=n;j++)
for(int k=j;k<=n;k++)
if(check(i,j,k))ans[i][j][k]=1;
scanf("%d",&m);
while(m--)
{
for(int i=0;i<3;i++)
scanf("%d",&q[i]);
sort(q,q+3);
if(ans[q[0]][q[1]][q[2]])
printf("Yes\n");
else
printf("No\n");
}
}
return 0;
}