hdu_5890_Eighty seven(bitset优化DP)

时间:2022-06-26 08:27:56

题目链接:hdu_5890_Eighty seven

题意:

50个数,10W个询问,每次问删掉第i,j,k个数后,是否存在一种选10个数和为87的方案,只需要输出 ’Yes’ 或者 ’No’

题解:

暴力:不同的询问大概2W个,每个暴力bitset DP,抠一抠能卡着过。优化1:先求出一组解,如果询问和解没交就是’Yes’,否则暴力,不同的询问大概1W个;优化2:先预处理出所有询问的答案,能方便的复用之前的DP数组,不用每次从头开始重新求。

以上是camp的题解

这里学习了一下bitset,把87*n优化到了10*n,然后可以卡过去

dp[i][j]表示选了i个数和为j的情况,dp[j]|=dp[j-1]<<a[i],j从10到1,这样就能优美的记录和的情况

 #include<bits/stdc++.h>
#define F(i,a,b) for(int i=a;i<=b;i++)
using namespace std; bitset<>dp[];
int ans[][][],a[],n,m,q[],T; bool check(int x,int y,int z)
{
F(i,,)dp[i].reset();
dp[][]=;
F(i,,n)if(i!=x&&i!=y&&i!=z)for(int j=;j>;j--)dp[j]|=dp[j-]<<a[i];
return dp[][];
} int main()
{
scanf("%d",&T);
while(T--)
{
memset(ans,,sizeof(ans));
scanf("%d",&n);
F(i,,n)scanf("%d",a+i);
F(i,,n)F(j,i,n)F(k,j,n)if(check(i,j,k))ans[i][j][k]=;
scanf("%d",&m);
while(m--)
{
F(i,,)scanf("%d",q+i);
sort(q,q+);
if(ans[q[]][q[]][q[]])puts("Yes");else puts("No");
}
}
return ;
}