题意:从n个数中选出若干个数使得和为给定值m,题目保证有解。
由于n最大为36,直接搜索2^36次方肯定TLE,但对于区间的一半即18,2 ^18则可以,故应该得想到把区间折半,先计算出前半区间各种组合情况的和,在枚举后半区间各种情况的和sum在前半区间中找m-sum是否存在。
法一:set+map
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<set>
#include<map>
using namespace std;
typedef long long ll;
ll a[50];
map<ll,int>mp;
set<ll>st;
int main()
{
int i,j;
int n;
ll m;
scanf("%d%lld",&n,&m);
for(i=0;i<n;i++)
{
scanf("%lld",&a[i]);
}
int p=n/2;
int q=n-p;
for(i=0;i<(1<<p);i++)//前半区间;
{
ll sum=0;
for(j=0;j<p;j++)
{
if((i>>j)&1)
{
sum+=a[j];
}
}
st.insert(sum);
mp[sum]=i;
}
for(i=0;i<(1<<q);i++)//后半区间;
{
ll sum=0;
for(j=0;j<q;j++)
{
if((i>>j)&1)
{
sum+=a[j+p];
}
}
if(st.find(m-sum)!=st.end())
{
int now=mp[m-sum];
for(j=0;j<p;j++)
{
printf("%d",(now>>j)&1);
}
for(j=0;j<q;j++)
{
printf("%d",(i>>j)&1);
}
}
}
return 0;
}
法二:map+string:
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <string>
#include <map>
using namespace std;
typedef long long ll;
ll a[50];
int main()
{
int i,j;
map<ll,string>mp1;
map<ll,string>mp2;
int n;
ll m;
scanf("%d%lld",&n,&m);
for(i=0;i<n;i++)
{
scanf("%lld",&a[i]);
}
int p=n/2;
int q=n-p;
for(i=0;i<(1<<p);i++)//前半段;
{
string s;
ll sum=0;//选择其中某些的和;
for(j=0;j<p;j++)
{
if((i>>j)&1)//这个是判断第j位选不选;
{
sum+=a[j];
s.push_back('1');
}
else
{
s.push_back('0');
}
}
mp1[sum]=s;
}
for(i=0;i<(1<<q);i++)
{
string s;
ll sum=0;
for(j=0;j<p;j++)
{
if((i>>j)&1)
{
sum+=a[j+p];
s.push_back('1');
}
else
{
s.push_back('0');
}
}
mp2[sum]=s;
if(mp1.count(m-sum))
{
cout<<mp1[m-sum]<<mp2[sum]<<endl;
break;
}
}
return 0;
}