[CEOI2015 Day2] 世界冰球锦标赛（折半搜索）

[CEOI2015 Day2] 世界冰球锦标赛

题目描述

译自 CEOI2015 Day2 T1「Ice Hockey World Championship」

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

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

输入格式

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

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

输出格式

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

样例 #1

样例输入 #1

5 1000
100 1500 500 500 1000

样例输出 #1

8

提示

样例解释

八种方案分别是：

• 一场都不看，溜了溜了
• 价格 $100$ 的比赛
• 第一场价格 $500$ 的比赛
• 第二场价格 $500$ 的比赛
• 价格 $100$ 的比赛和第一场价格 $500$ 的比赛
• 价格 $100$ 的比赛和第二场价格 $500$ 的比赛
• 两场价格 $500$ 的比赛
• 价格 $1000$ 的比赛

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

数据组号	$1-2$	$3-4$	$5-7$	$8-10$
$N\le$	$10$	$20$	$40$	$40$
$M\le$	$10^6$	$10^{18}$	$10^6$	$10^{18}$

思路

• 因为这道题的 $M$ 很大，如果我们用背包的话，会直接MLE。
• 那我们该怎么做呢？
• 我们此时看到 $N$ 的范围很小，我们就可以用搜索的背包来做，我们就大胆的用爆搜。好，TLE。此时我们就考虑折半搜索。（因为直接爆搜是 $O(2^{40})$）
• 折半搜索思路：我们现在可以将所有比赛的序列分成两个部分，分别对这两部分进行搜索，然后用数组来存储可能的方案与花费，在排序使其有序，最后用在合并，二分搜索查找可行解，统计答案累加，得出最终的答案。
• 折半搜索都是一个dfs是用来搜索答案的，然后第二个搜索时用来累加答案的，就是减少搜索次数。

具体是什么样子的，我们可以看代码：

代码

//搜索+背包（暴搜的背包）

#include<iostream>
#include<algorithm>
#include<cstring>

using namespace std;

const int N = 45;

typedef long long LL;

LL n,m;
LL b;
LL h[(1<<22)+5];//所有可行方案先放在这里，等等给后面用
LL w[N],ans;
int cnt;

void dfs1(int u,LL val){
    if(u==b+1){
        h[++cnt]=val;//存储答案
        return;
    }
    
    dfs1(u+1,val);
    
    if(val+w[u]<=m){
        dfs1(u+1,val+w[u]);
    }
}

void dfs2(int u,LL val){
    if(u==n+1){
        //二分查找找到花费比剩下钱还少的方案数,累加答案 
        ans+=upper_bound(h+1,h+cnt+1,m-val)-h-1;
        return;
    }
    
    dfs2(u+1,val);
    
    if(val+w[u]<=m){
        dfs2(u+1,val+w[u]);
    }
}

int main(){
    cin>>n>>m;
    
    for(int i=1;i<=n;i++)cin>>w[i];
    
    b=min((n>>1)+2,n-1);
    
    sort(w+1,w+1+n);
    
    reverse(w+1,w+1+n);
    
    dfs1(1,0);
    
    sort(h+1,h+cnt+1);
    
    dfs2(b+1,0);
    
    //2^40特别大，因此我们折半搜索
    cout<<ans;
    
    return 0;
}