第一次写博客...本场比赛爆0了...写写这一道签到题
给定最多36个数字和sum 问从中选出一些数字可以组成sum 问选出哪些数字可以组成 输出01串 0代表不选1代表选
思路 折半搜索 将n分为前后两部分 前半部分枚举所有情况 这个地方枚举利用了2进制 i从0(全不选) 一直枚举到(1<<n/2)-1(即01111111...)就是前半部分全选 枚举完之后将前半部分所有的和存入set和map(貌似可以用unorder_map,好像更快) set用于后半部分求和的时候搜索用 map用来寻找和这个和所对应的i(枚举的方法)
之后再看后半部分 同理 枚举情况 不过这次是枚举完一种情况就去set里搜一下 如果搜得到 就从Map中直接调出第一种枚举的方法 之后输出就可以了
#include<algorithm> #include<iostream> #include<vector> #include<cstdio> #include<cstring> #include<cmath> #include<string> #include<ctime> #include<map> #include<stack> #include<queue> #include<set> #include<cstring> using namespace std; const int INF = 0x3f3f3f3f; const int maxn = 1e5 + 5; typedef long long ll; typedef pair<ll, ll> pi; int n, p, q; set<ll> s; map<ll, int>f; ll a[36], sum; int main() { ios::sync_with_stdio(0); cin.tie(0); cout.tie(0); cin >> n; p = n / 2;//分两半 q = n - p;//另一半 cin >> sum; for (int i = 0; i < n; i++) { cin >> a[i]; } for (int i = 0; i < (1 << p); i++) {//比如100000000代表一半 从0开始枚举一半的所有状态 ll ret = 0; for (int j = 0; j < p; j++) { if (i & (1 << j)) {//如果i上这位为1 那么加上去 ret += a[j]; } } s.insert(ret);//存进set f[ret] = i; } for (int i = 0; i < (1 << q); i++) {//枚举另一半 ll ret = 0; for (int j = 0; j < q; j++) { if (i & (1 << j)) { ret += a[j + p];//后半部分 } } //每算完一次 就去s中查找一次 if (s.find(sum - ret) != s.end()) {//如果找到了 int x = f[sum - ret]; for (int j = 0; j < p; j++) {//输出前半部分 printf("%d", (bool)(x & (1 << j))); } for (int j = 0; j < q; j++) { printf("%d", (bool)(i & (1 << j))); } puts(""); break; } } return 0; }
Comments | NOTHING