洛谷 P4799 [CEOI2015 Day2] 世界冰球锦标赛

题目描述

译自 CEOI2015 Day2 T1「Ice Hockey World Championship」

今年的世界冰球锦标赛在捷克举行。Bobek 已经抵达布拉格，他不是任何团队的粉丝，也没有时间观念。他只是单纯的想去看几场比赛。如果他有足够的钱，他会去看所有的比赛。不幸的是，他的财产十分有限，他决定把所有财产都用来买门票。

给出 Bobek 的预算和每场比赛的票价，试求：如果总票价不超过预算，他有多少种观赛方案。如果存在以其中一种方案观看某场比赛而另一种方案不观看，则认为这两种方案不同。

输入格式

第一行，两个正整数 NN 和 M(1 \leq N \leq 40,1 \leq M \leq 10^{18})M(1≤N≤40,1≤M≤1018)，表示比赛的个数和 Bobek 那家徒四壁的财产。

第二行，NN 个以空格分隔的正整数，均不超过 10^{16}1016，代表每场比赛门票的价格。

输出格式

输出一行，表示方案的个数。由于 NN 十分大，注意：答案 \le 2^{40}≤240。

输入输出样例

输入 #1复制

5 1000
100 1500 500 500 1000

输出 #1复制

8

说明/提示

样例解释

八种方案分别是：

• 一场都不看，溜了溜了
• 价格 100100 的比赛
• 第一场价格 500500 的比赛
• 第二场价格 500500 的比赛
• 价格 100100 的比赛和第一场价格 500500 的比赛
• 价格 100100 的比赛和第二场价格 500500 的比赛
• 两场价格 500500 的比赛
• 价格 10001000 的比赛

有十组数据，每通过一组数据你可以获得 10 分。各组数据的数据范围如下表所示：

数据组号	1-2	3-4	5-7	8-10
N <=	10	20	40	40
M <=	10^6	10^18	10^6	10^18

代码如下：

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
LL a[50],n,m;		//n：表示比赛个数；m：表示Bobek那家徒四壁的财产。 
vector<LL>fir,sec;
void dfs1(int k,LL sum){
	if(k==n/2+1){
		fir.push_back(sum);
		return;
	}
	dfs1(k+1,sum);
	dfs1(k+1,sum+a[k]);
}
void dfs2(int k,LL sum){
	if(k==n+1){
		sec.push_back(sum);
		return;
	}
	dfs2(k+1,sum);
	dfs2(k+1,sum+a[k]);
}
int main(){
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	dfs1(1,0);
	dfs2(n/2+1,0);
	sort(fir.begin(),fir.end());
	LL ans=0;
	for(int i=0;i<(int)sec.size();i++){
		ans+=upper_bound(fir.begin(),fir.end(),m-sec[i])-fir.begin();
	}
	cout<<ans;
	return 0;
}

由于数据比较庞大，所以不能直接搜索，可以将数据分成两部分，分别搜索。（注意：使用到了STL模板库）