Codeforces Round #297 (Div. 2) E

Anya and Cubes

        这个数据范围，很明显的折半搜索，还是不难写的。分别从前往后和从后往前dfs。从前往后搜到中间，结果存在一个map里，从后往前搜到中间，查询map。

        另外做这题有个小发现。就是你在查询map[x]的值的时候，虽然不是给它“赋值”，但是如果本来x不存在的话，还是会把x插入到map里面去，如果只是为了查询是否存在，最好用map.count()。

#include <bits/stdc++.h>

using namespace std;

#define ll long long

int a[30];
int n,k;
ll S;
int mid;
map<ll,ll> mp[30];
ll fac[20];

//当前位置 当前和 已选个数 
void dfs1(int pos,ll sum,int cnt){
	if(sum>S)return;
	if(cnt>k)return;
	if(pos>mid){
		mp[cnt][sum]++;
		return;
	}
	
	//不选 
	dfs1(pos+1,sum,cnt);
	//选 
	dfs1(pos+1,sum+a[pos],cnt); 
	//变阶乘 
	if(a[pos]<=18){
		dfs1(pos+1,sum+fac[a[pos]],cnt+1);
	}
}

ll ans;

void dfs2(int pos,ll sum,int cnt){
	if(sum>S)return;
	if(cnt>k)return;
	if(pos==mid){
		for(int i=0;cnt+i<=k;i++){
			if(mp[i].count(S-sum)){
				ans+=mp[i][S-sum];
			}
		}
		return;
	}
	//不选 
	dfs2(pos-1,sum,cnt);
	//选 
	dfs2(pos-1,sum+a[pos],cnt); 
	//变阶乘 
	if(a[pos]<=18){
		dfs2(pos-1,sum+fac[a[pos]],cnt+1);
	}
	
}

void init(){
	ans=0;
	mid=n/2+1;
	fac[0]=1;
	for(int i=1;i<=18;i++){
		fac[i]=fac[i-1]*i;
	}
}

int main(){
	cin>>n>>k>>S;
	init();
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	sort(a+1,a+n+1);
	dfs1(1,0,0);
	dfs2(n,0,0);
	
	cout<<ans<<endl;
	
	return 0;
}