码迷,mamicode.com
首页 > 其他好文 > 详细

Anya and Cubes CodeForces - 525E (双端搜索)

时间:2019-04-02 00:21:41      阅读:156      评论:0      收藏:0      [点我收藏+]

标签:cond   超过   复杂   cube   font   algo   typedef   red   second   

大意: 给定$n$元素序列$a$, 可以任选不超过$k$个$a_i$变换为$a_i!$, 求变换后任选若干元素和为S的方案数.

 

分成两块暴搜, 复杂度$O(3^{\frac{n}{2}})$

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <unordered_map>
#define REP(i,a,n) for(int i=a;i<=n;++i)
using namespace std;
typedef long long ll;

int n, k, a[30];
ll S, fac[30];
unordered_map<ll,int> x[30], y[30];

void dfs(int d, int mx, int kk, ll num, unordered_map<ll,int> *x) {
	if (num>S||kk>k) return;
	if (d==mx) return ++x[kk][num],void();
	dfs(d+1,mx,kk,num+a[d],x);
	if (a[d]<20) dfs(d+1,mx,kk+1,num+fac[a[d]],x);
	dfs(d+1,mx,kk,num,x);
}

int main() {
	fac[0] = 1;
	REP(i,1,19) fac[i]=fac[i-1]*i;
	cin>>n>>k>>S;
	REP(i,0,n-1) cin>>a[i];
	dfs(0,n/2,0,0,x), dfs(n/2,n,0,0,y);
	ll ans = 0;
	REP(i,0,k) for (auto &&it:x[i]) {
		auto &&t = it.second;
		auto &&num = S-it.first;
		REP(j,0,k-i) ans += t*y[j][num];
	}
	printf("%lld\n", ans);
}

 

Anya and Cubes CodeForces - 525E (双端搜索)

标签:cond   超过   复杂   cube   font   algo   typedef   red   second   

原文地址:https://www.cnblogs.com/uid001/p/10639966.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!