暴搜WA on test4 求调
查看原帖
暴搜WA on test4 求调
1285691
yinbe2楼主2024/4/18 21:20
#include<iostream>
#include<map>
using namespace std;
int n,a[30],k;
long long s,ans;
map<int,long long>f;
map<pair<long long,int>,bool>mp;
map<long long,bool>vis;
int mid;
void dfs_left(int t,long long sum,int cnt)
{
	if(t>mid)
	{
		mp[{sum,cnt}]++;
		vis[sum]=true;
		return;
	}
	dfs_left(t+1,sum,cnt);
	if(sum+a[t]<=s)dfs_left(t+1,sum+a[t],cnt);
	if(a[t]<=20&&cnt<=k&&sum+f[a[t]]<=s)dfs_left(t+1,sum+f[a[t]],cnt+1);
}
long long _count(long long sum,int cnt)
{
	long long ans=0;
	for(int i=0;i<=cnt;i++)ans+=mp[{sum,i}];
	return ans;
}
void dfs_right(int t,long long sum,int cnt)
{
	if(t>n)
	{
		if(vis.find({s-sum})!=vis.end())ans+=_count(s-sum,k-cnt);
		return;
	}
	dfs_right(t+1,sum,cnt);
	if(sum+a[t]<=s)dfs_right(t+1,sum+a[t],cnt);
	if(a[t]<=20&&cnt<=k&&sum+f[a[t]]<=s)dfs_right(t+1,sum+f[a[t]],cnt+1);
}
int main()
{
	f[1]=1;
	for(int i=2;i<=20;i++)f[i]=f[i-1]*i;
	scanf("%d%d%lld",&n,&k,&s);
	mid=(n>>1);
	for(int i=1;i<=n;i++)scanf("%d",&a[i]);
	dfs_left(1,0,0);
	dfs_right(mid+1,0,0);
	printf("%lld",ans);
	return 0;
}
2024/4/18 21:20
加载中...