选数中途相遇搜索

Posted 2020-10-29 EZ_WYC

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了选数中途相遇搜索相关的知识，希望对你有一定的参考价值。

题目

思路

就是个搜索啊

代码

#include <iostream>
#include <cstring>
#include <cstdlib>
#include <vector>
#include <set>
#include <algorithm>
#include <cstdio>

using namespace std;

const int N = 25;

int a[N];

int n;

struct Data
{   int val, status;
    bool operator < (const Data & rhs) const { return val < rhs.val; }
} num[1000010];
int tot;

void dfs_1(int x, int sum, int S)
{   if(x == n/2+1)
        num[++tot] = (Data){sum, S};
    else
    {   dfs_1(x+1, sum, S);
        dfs_1(x+1, sum-a[x], S|(1<<x));
        dfs_1(x+1, sum+a[x], S|(1<<x));
    }
}

bool vis[2097153];

void dfs_2(int x, int sum, int S)
{   if(x == n+1)
    {   int l = lower_bound(num+1, num+1+tot, (Data){-sum, 0}) - num;
        int r = upper_bound(num+1, num+1+tot, (Data){-sum, 0}) - num;
        for(int i=l; i<r; i++)
            vis[S|num[i].status] = 1; //大神犇tzw说这样可能会T
    }
    else
    {   dfs_2(x+1, sum, S);
        dfs_2(x+1, sum-a[x], S|(1<<x));
        dfs_2(x+1, sum+a[x], S|(1<<x));
    }
}

int main()
{   scanf("%d", &n);
    for(int i=1; i<=n; i++)
        scanf("%d", &a[i]);
    dfs_1(1, 0, 0);
    sort(num+1, num+1+tot);
    dfs_2(n/2+1, 0, 0);
    int ans = 0;
    for(int i=0; i<=2097152; i++) if(vis[i]) ans++; //千万不要用set去重， 特别慢
    printf("%d\n", ans-1);
    return 0;
}