BZOJ 4555 [Tjoi2016&Heoi2016]求和 NTT

題目在這裏呀

題解

題目給你的那個遞推式好廢啊(沒有一點用還是要百度通項)。
知道第二類斯特林數時只要帶入原式展開可以發現卷積式,對它做NTT即可。
詳細見學姐的博客
這篇博客講的很清晰很具體,不會展開的話就看這篇吧qaq(不是我懶,是因爲我也是看了這篇纔會的啦ww)

//Suplex
#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#include <cmath>
#define ll long long
#define N 100000+1000
using namespace std;
const int mod=998244353;
//f(i)=(1/i!)*sigma(i^n)
//g(i)=(1/i!)*(-1)^i
//F(j)=(j!)*2^j*sigma(f(j-k)*g(k))

int n,m,L;
ll ans,now,f[N+N+N],g[N+N+N],fac[N],inv[N+N+N],invv[N],fac2[N],w[N+N+N];

inline ll power(ll a,int b)
{
    ll mul=1;
    while(b){
        if(b & 1) mul=(mul*a) % mod;
        a=a*a % mod;
        b>>=1;
    }
    return mul;
}

inline ll mo(ll xx)
{
    if(xx>mod) return xx-mod;
    if(xx<0) return xx+mod;
    return xx;
}

void NTT(ll *a,int opt)
{
    for(int i=0,j=0;i<L;i++){
        if(i>j) swap(a[i],a[j]);
        for(int k=L>>1;(j ^= k)<k;k>>=1);
    }
    for(int len=2;len<=L;len<<=1){
        int l=len>>1;
        ll W=power(3,(mod-1)/len);
        for(int i=1;i<l;i++) w[i]=w[i-1]*W % mod;
        for(int i=0;i<L;i+=len)
            for(int j=0;j<l;j++){
                ll x=a[i+j],y=w[j]*a[i+j+l] % mod;
                a[i+j]=mo(x+y);a[i+j+l]=mo(x-y);
            }
    }
    if(opt==-1)
    {
        for(int i=1;i<L/2;i++) swap(a[i],a[L-i]);
    //  for(int i=0;i<L;i++) a[i]=a[i]*inv[L] % mod;
    }
}

int main()
{
    scanf("%d",&n);
    m=n<<1;w[0]=1;
    for(L=1;L<=m;L<<=1);
    fac[0]=1ll;
    for(int i=1;i<=n;i++) fac[i]=(fac[i-1]*(ll)i) % mod;
    fac2[0]=1ll;
    for(int i=1;i<=n;i++) fac2[i]=(fac2[i-1]+fac2[i-1]) % mod;
    inv[1]=1ll;
    for(int i=2;i<=L;i++) inv[i]=(ll)(mod-mod/i)*inv[mod % i] % mod;
    invv[0]=1ll;
    for(int i=1;i<=n;i++) invv[i]=invv[i-1]*inv[i] % mod;
    now=-1;f[0]=g[0]=1ll;
    for(int i=1;i<=n;i++){
        g[i]=now*invv[i] % mod;
        if(i==1) f[i]=(ll)n+1;
        else f[i]=invv[i]*(power(i,n+1)-1) % mod * inv[i-1] % mod;
        now=-now;
    }
    NTT(f,1);NTT(g,1);
    for(int i=0;i<=L;i++) f[i]=f[i]*g[i] % mod;
    NTT(f,-1);
    for(int i=0;i<=n;i++)
        ans=(ans+fac[i]*fac2[i] % mod * f[i] % mod * inv[L] % mod) % mod;
    ans=(ans+mod) % mod;
    printf("%lld\n",ans);
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章