bzoj 4555: [Tjoi2016&Heoi2016]求和

题意:

i=0nj=0iS(i,j)2jj!

题解:

这个式子的意义是将i个不同的物品,放入j个不同的盒子,每个盒子有两种状态的方案数。
g[n]=ni=0S(n,i)2ii!
可以得到 g 的递推式:g[n]=ni=12Cing(ni)
于是:

g[n]n!=i=1n2i!g(ni)(ni)!

得到卷积形式: G(x)=G(x)H(x)+1
G=(1H)1

code:

#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<cstring>
#define LL long long
using namespace std;
LL p=998244353,g=3;
LL pow(LL a,LL b,LL mod)
{
    LL ans=1;
    while(b)
    {
        if(b&1) ans=ans*a%mod;
        a=a*a%mod;b>>=1;
    }
    return ans;
}
int bin[400010];
void ntt(LL *a,int n,int op)
{
    for(int i=0;i<n;i++) bin[i]=(bin[i>>1]>>1)|((i&1)*(n>>1));
    for(int i=0;i<n;i++) if(i<bin[i]) swap(a[i],a[bin[i]]);
    for(int i=1;i<n;i<<=1)
    {
        LL wn=pow(g,op==1?(p-1)/(2*i):p-1-(p-1)/(2*i),p),t,w;
        for(int j=0;j<n;j+=i<<1)
        {
            w=1;
            for(int k=0;k<i;k++)
            {
                t=w*a[i+j+k]%p;w=w*wn%p;
                a[i+j+k]=(a[j+k]-t+p)%p;a[j+k]=(a[j+k]+t)%p;
            }
        }
    }
    if(op==-1)
    {
        LL Inv=pow(n,p-2,p);
        for(int i=0;i<n;i++) a[i]=a[i]*Inv%p; 
    }
}
void pol_inv(int deg,LL *a,LL *b,LL *tmp)
{
    if(deg==1) {b[0]=pow(a[0],p-2,p);return;}
    pol_inv((deg+1)>>1,a,b,tmp);
    int n=1;while(n<deg<<1) n<<=1;
    copy(a,a+deg,tmp);fill(tmp+deg,tmp+n,0);
    ntt(tmp,n,1);ntt(b,n,1);
    for(int i=0;i<n;i++)
    {
        b[i]=(2-b[i]*tmp[i]%p)*b[i]%p;
        if(b[i]<0) b[i]+=p;
    }
    ntt(b,n,-1);fill(b+deg,b+n,0);
}
int n;
LL f[100010],inv[100010];
LL a[400010],b[400010],c[400010];
int main()
{
    scanf("%d",&n);
    f[0]=f[1]=inv[0]=inv[1]=1;
    for(int i=2;i<=n;i++) inv[i]=(LL)(p-p/i)*inv[p%i]%p;
    for(int i=2;i<=n;i++) inv[i]=inv[i-1]*inv[i]%p,f[i]=f[i-1]*i%p;
    for(int i=1;i<=n;i++) a[i]=(-2*inv[i]+p)%p;a[0]=1;
    pol_inv(n+1,a,b,c);
    LL ans=0;
    for(int i=0;i<=n;i++) ans=(ans+b[i]*f[i]%p)%p;
    printf("%lld",ans);
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值