折半枚举
传统的双向搜索
https://zh.wikipedia.org/wiki/%E5%8F%8C%E5%90%91%E6%90%9C%E7%B4%A2
折半枚举与传统的双向搜索不同,不过思想来源于传统的双向搜索。
情况
当数据量特别大的时候,无法枚举所有的组合,只够枚举一半的组合,这种情况下我们就可以用折半枚举解决问题。
做法
将原序列分成两半,分别枚举两个部分的组合,然后根据题目给出的条件,(一般用二分)找到两个部分中要取的元素(将它们合并起来,得到真实的答案)。
例题
19牛客多校第九场 D题
https://ac.nowcoder.com/acm/contest/889/D
代码
第一次做折半枚举的题目,代码写的比较长。
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
ll n, s;
ll num[40];
int cnt1 = 0;
int cnt2 = 0;
struct node
{
ll sum;
ll consist;
bool operator < (const node &p)const{
return sum < p.sum;
}
}a[1<<20], b[1<<20];
bool cmp(struct node a, struct node b)
{
return a.sum < b.sum;
}
int dfs1(int i, int e, ll sum, ll consist)
{
if(i == e)
{
a[cnt1].sum = sum;
a[cnt1++].consist = consist;
return 0;
}
if(i + 1 <= e)//不取
{
dfs1(i + 1, e, sum, consist);
dfs1(i + 1, e, sum + num[i + 1], consist + (1ll << (i + 1)));//反着放进去,所以输出的时候要从后往前输出
}
return 0;
}
int dfs2(int i, int e, ll sum, ll consist)
{
if(i == e)
{
b[cnt2].sum = sum;
b[cnt2++].consist = consist;
return 0;
}
if(i + 1 <= e)
{
dfs2(i + 1, e, sum, consist);
dfs2(i + 1, e, sum + num[i + 1], consist + (1ll << (i + 1)));
}
return 0;
}
int main()
{
scanf("%d %lld", &n, &s);
for(int i = 0; i < n; ++i)
{
scanf("%lld", &num[i]);
}
if(n <= 18)
{
dfs1(-1, n, 0, 0);
}
else if(n > 18)
{
dfs1(-1, 17, 0, 0);
dfs2(17, n - 1, 0, 0);
}
if(n <= 18)
{
sort(a, a + cnt1, cmp);
for(int i = 1; i <= cnt1; ++i)
{
if(a[i].sum == s)
{
for(int j = 1; j <= n; ++j)
{
printf("%lld", a[i].consist % 2);
a[i].consist /= 2;
}
return 0;
}
}
}
else if(n > 18)
{
sort(a, a + cnt1, cmp);
sort(b, b + cnt2, cmp);
for(int i = 1; i <= cnt1; ++i)
{
int pos = lower_bound(b , b + cnt2, (node){s - a[i].sum, 0}) - b;
if(a[i].sum + b[pos].sum == s)
{
//printf("b[]consist:%lld\n",b[pos].consist);
int j;
for(j = 1; j <= 18; ++j)
{
printf("%lld", a[i].consist % 2);
a[i].consist /= 2;
}
b[pos].consist >>= 18;
for(;j <= n; ++j)
{
printf("%lld", b[pos].consist % 2);
b[pos].consist /= 2;
}
return 0;
}
}
}
return 0;
}
参考来源
白书P160
博客