ACdream 1726 A Math game (dfs+二分)

时间:2023-03-08 22:09:51

http://acdream.info/problem?pid=1726

官方题解:http://acdream.info/topic?tid=4246

求n个数里面能不能选一些数出来让它们的和等于k。

因为k很大,不能用背包,但是n很小,最大为40,所以拆成了2部分,之后最大为2^20次方<1050000;每次枚举前一半的和,然后用数组存储,然后得到一个总和减去后一半的差用二分查找。

 #include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std; int n,h,n1,n2,a[],b[],c[],cnt;
bool flag=; bool check(int x)
{
int l=,r=cnt-;
while(l<=r)
{
int mid=(l+r)>>;
if(c[mid]==x) return ;
else if(c[mid]>x) r=mid-;
else l=mid+;
}
return ;
}
void dfs(int sum,int deep,int m,int on)
{
if(flag) return; //剪枝
if(deep==m) //到最后一个数
{
if(on) c[cnt++]=sum; //第一次记录 和
else flag=check(h-sum); //第二次查找
return;
}
for(int i=;i<;i++)
{
sum+=b[deep]*i; //累加和
if(sum>h) return; //剪枝
dfs(sum,deep+,m,on);
}
}
int main()
{
//freopen("a.txt","r",stdin);
while(~scanf("%d%d",&n,&h))
{
cnt=;
flag=;
for(int i=;i<n;i++)
scanf("%d",&a[i]);
n1=n>>;n2=n-n1; //分成两半
for(int i=n1;i<n;i++)
b[i-n1]=a[i]; //后一半数用b存储
dfs(,,n2,); //枚举所有可能的和
sort(c,c+cnt); //对得到的数组进行排序 便于二分查找
for(int i=;i<n1;i++)
b[i]=a[i]; //得到前一半的和
dfs(,,n1,); //枚举所有的和
if(flag) puts("Yes");
else puts("No");
}
return ;
}