预处理,$01$背包,$bitset$优化。
可以预处理出每一种询问的答案,用$01$背包计算答案,$dp[i][j][k]$表示,前$i$个数字中,选择了$j$个,能否凑出$k$这个数字,可以开成$bitset<90>dp[55][12]$,第三维$bitset$位运算优化。
$HDU$不稳,有时超时,有时通过。
#pragma comment(linker, "/STACK:1024000000,1024000000") #include<cstdio> #include<cstring> #include<cmath> #include<algorithm> #include<vector> #include<map> #include<set> #include<queue> #include<stack> #include<bitset> #include<iostream> using namespace std; typedef long long LL; const double pi=acos(-1.0),eps=1e-6; void File() { freopen("D:\in.txt","r",stdin); freopen("D:\out.txt","w",stdout); } template <class T> inline void read(T &x) { char c=getchar(); x=0; while(!isdigit(c)) c=getchar(); while(isdigit(c)) {x=x*10+c-'0'; c=getchar();} } const int maxn=60; int T,n,s[60]; bool f[maxn][maxn][maxn],g[maxn][maxn][maxn]; bool get(int a,int b,int c) { bitset<90>dp[maxn][12]; int p[60], sz=0; for(int i=1;i<=n;i++) { if(s[i]>87) continue; if(i==a) continue; if(i==b) continue; if(i==c) continue; p[sz++]=s[i]; } if(sz==0) return 0; dp[0][1][p[0]]=1; for(int i=1;i<sz;i++) { dp[i][1]|=dp[i-1][1]; dp[i][1][p[i]]=1; for(int j=2;j<=10;j++) { dp[i][j]|=dp[i-1][j]; dp[i][j]|=dp[i-1][j-1]<<p[i]; } if(dp[i][10][87]==1) return 1; } return 0; } int main() { scanf("%d",&T); for(int i=1;i<=T;i++) { memset(f,0,sizeof f); memset(g,0,sizeof g); scanf("%d",&n); for(int i=1;i<=n;i++) scanf("%d",&s[i]); int Q; scanf("%d",&Q); while(Q--) { int op[3]; scanf("%d%d%d",&op[0],&op[1],&op[2]); sort(op,op+3); if(g[op[0]][op[1]][op[2]]) { if(f[op[0]][op[1]][op[2]]==0)printf("No "); else printf("Yes "); } else { f[op[0]][op[1]][op[2]]=get(op[0],op[1],op[2]); if(f[op[0]][op[1]][op[2]]==0)printf("No "); else printf("Yes "); g[op[0]][op[1]][op[2]]=1; } } } return 0; }