【Codeforces Round #297 (Div. 2)】Codeforces 525E Anya and Cubes

Anya loves to fold and stick. Today she decided to do just that.

Anya has n cubes lying in a line and numbered from 1 to n from left to
right, with natural numbers written on them. She also has k stickers
with exclamation marks. We know that the number of stickers does not
exceed the number of cubes.

Anya can stick an exclamation mark on the cube and get the factorial
of the number written on the cube. For example, if a cube reads 5,
then after the sticking it reads 5!, which equals 120.

You need to help Anya count how many ways there are to choose some of
the cubes and stick on some of the chosen cubes at most k exclamation
marks so that the sum of the numbers written on the chosen cubes after
the sticking becomes equal to S. Anya can stick at most one
exclamation mark on each cube. Can you do it?

Two ways are considered the same if they have the same set of chosen
cubes and the same set of cubes with exclamation marks. Input

The first line of the input contains three space-separated integers n,
k and S (1 ≤ n ≤ 25, 0 ≤ k ≤ n, 1 ≤ S ≤ 1016) — the number of cubes
and the number of stickers that Anya has, and the sum that she needs
to get.

The second line contains n positive integers ai (1 ≤ ai ≤ 109) — the
numbers, written on the cubes. The cubes in the input are described in
the order from left to right, starting from the first one.

Multiple cubes can contain the same numbers. Output

Output the number of ways to choose some number of cubes and stick
exclamation marks on some of them so that the sum of the numbers
became equal to the given number S.

每个物品有三种决策,直接暴力枚举O(3^n)会超时,但是我们发现O(3^(n/2))不会超时,于是可以采用中途相遇法。
先暴力枚举一半的决策,把对应的值、使用次数和方案数存到map里,然后再枚举另一半,在map中查找。
复杂度O(3^(n/2)log(3^(n/2)))=O(3^(n/2)n)。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<map>
using namespace std;
#define LL long long
map<LL,int> m[30];
int n,k;
LL s,a[30],fact[20],ans;
void dfs1(LL sum,int p,int now)
{
    if (p==n/2+1)
    {
        if (m[now].count(sum)) m[now][sum]++;
        else m[now][sum]=1;
        return;
    }
    dfs1(sum,p+1,now);
    if (a[p]<=s-sum) dfs1(sum+a[p],p+1,now);
    if (now<k&&a[p]<=19&&fact[a[p]]<=s-sum) dfs1(sum+fact[a[p]],p+1,now+1);
}
void dfs2(LL sum,int p,int now)
{
    if (p==n+1)
    {
        for (int i=0;i+now<=k;i++)
          if (m[i].count(s-sum))
            ans+=m[i][s-sum];
        return;
    }
    dfs2(sum,p+1,now);
    if (a[p]<=s-sum) dfs2(sum+a[p],p+1,now);
    if (now<k&&a[p]<=19&&fact[a[p]]<=s-sum) dfs2(sum+fact[a[p]],p+1,now+1);
}
int main()
{
    int i,j;
    LL x,y,z;
    fact[0]=1;
    for (i=1;i<=19;i++)
      fact[i]=fact[i-1]*i;
    scanf("%d%d%I64d",&n,&k,&s);
    for (i=1;i<=n;i++)
      scanf("%I64d",&a[i]);
    dfs1(0,1,0);
    dfs2(0,n/2+1,0);
    printf("%I64d\n",ans);
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值