2019牛客暑期多校（D--Knapsack Cryptosystem）(折半查找）-CSDN博客

本文链接：https://blog.csdn.net/weixin_44756457/article/details/99684505

本文介绍了解决子集求和问题的一种高效算法，通过将问题分解为两半并利用set和map数据结构，避免了直接搜索的高时间复杂度。前半区间的所有可能组合被计算并存储，然后在后半区间中查找与目标和匹配的值，实现了快速求解。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意：从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;
}