19牛客多校第九场 D题 折半枚举

题意

给你个序列和数字S，问你哪些数的和等于S，要求输出01串，表示对应位置的数字取与不取。

思路

最多有36个数字，传统的搜索，最多 2³⁶ 次操作，加上剪枝也会TLE。
这里用折半枚举（思想来源于传统的双向搜索）就可以解决问题。
把序列分成两个部分，分别枚举这两个部分的组合，然后对于其中一个部分的和，看另一个部分有没有这样一种，使得它跟第一个部分之和等于题目给的S，这里用lower_bound即可。

代码

#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;
}