【TJOI2019]歌う、ダンス、ラップ、バスケットボール(生成機能、組合せ論、NTT)

みなさ今年のピットを埋めます。

プログラムの設定(最も正確な)によると、問題にWWJソリューション、\(S \)鶏の数はあなたがとても美しいグループだ(必ずしも法的に)\()S(W \)

答えは\(\ SUM \ limits_ {S} [W(S)= 0] \)

用二项式定理:\(\和\ limits_ {S} [W(S)= 0] = \和\ limits_ {S}(1-1)^ {W(S)} = \和\ limits_ {S } \和\ limits_ {iはGE 0 \}( - 1)^ iはbinomを\ {W(S)} {I} = \和\ limits_ {iはGE 0 \}( - 1)^ iは\和\ limits_ { S} \ binom {W(S)} {I} \)

その合計数は、すべてのプログラムのために必要とされる背景には、あなたはとても美しいグループだ鶏から番号を選択\(私は\)のグループやプログラムの数を。

列挙基は、位置を選択した場合の合計\(\ binom {N-3I } {I} \) プログラムの種類。(のバインディング、合計\(N-3I \)個人、\(私は\)あなたはとても美しいです個人鶏)

順列スキームの残りの部分、それぞれの数字、愛する列挙人\(\ SUM \ limits_ {W \ルAI} \ SUM \ limits_ {X \}ルBI \ SUM \のlimits_ {Y \ CI}ルを\ SUM \ limits_ {Z \ルDI} [W + X + Y + Z = N - 4I] \ FRAC {(N - 4I)!W} \){X Y Z!!}

それぞれについて、明らかに畳み込み\(私は\)再びNTTうまくそれを行うには。

時間複雑\(O(N-2 ^ \ n型ログ)\)

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> PII;
const int maxn=2222,mod=998244353;
#define MP make_pair
#define PB push_back
#define lson o<<1,l,mid
#define rson o<<1|1,mid+1,r
#define FOR(i,a,b) for(int i=(a);i<=(b);i++)
#define ROF(i,a,b) for(int i=(a);i>=(b);i--)
#define MEM(x,v) memset(x,v,sizeof(x))
inline ll read(){
    char ch=getchar();ll x=0,f=0;
    while(ch<'0' || ch>'9') f|=ch=='-',ch=getchar();
    while(ch>='0' && ch<='9') x=x*10+ch-'0',ch=getchar();
    return f?-x:x;
}
int n,a,b,c,d,fac[maxn],invfac[maxn],A[maxn],B[maxn],C[maxn],D[maxn],F[maxn],lim,l,rev[maxn],ans;
int qpow(int a,int b){
    int ans=1;
    for(;b;b>>=1,a=1ll*a*a%mod) if(b&1) ans=1ll*ans*a%mod;
    return ans;
}
void init(int upr){
    for(lim=1,l=0;lim<upr;lim<<=1,l++);
    FOR(i,0,lim-1) rev[i]=(rev[i>>1]>>1)|((i&1)<<(l-1)); 
}
void NTT(int *A,int tp){
    FOR(i,0,lim-1) if(i<rev[i]) swap(A[i],A[rev[i]]);
    for(int i=1;i<lim;i<<=1)
        for(int j=0,Wn=qpow(3,mod-1+tp*(mod-1)/(i<<1));j<lim;j+=i<<1)
            for(int k=0,w=1;k<i;k++,w=1ll*w*Wn%mod){
                int x=A[j+k],y=1ll*A[i+j+k]*w%mod;
                A[j+k]=(x+y)%mod;
                A[i+j+k]=(x-y+mod)%mod;
            }
    if(tp==-1){
        int linv=qpow(lim,mod-2);
        FOR(i,0,lim-1) A[i]=1ll*A[i]*linv%mod;
    }
}
int calc(int x){
    init(a+b+c+d-4*x);
    FOR(i,0,lim-1) A[i]=B[i]=C[i]=D[i]=0;
    FOR(i,0,a-x) A[i]=invfac[i];
    FOR(i,0,b-x) B[i]=invfac[i];
    FOR(i,0,c-x) C[i]=invfac[i];
    FOR(i,0,d-x) D[i]=invfac[i];
    NTT(A,1);NTT(B,1);NTT(C,1);NTT(D,1);
    FOR(i,0,lim-1) F[i]=1ll*A[i]*B[i]%mod*C[i]%mod*D[i]%mod;
    NTT(F,-1);
//  printf("calc(%d),f=%d,fac=%d\n",x,F[n-4*x],fac[n-4*x]);
    return 1ll*fac[n-4*x]*F[n-4*x]%mod; 
}
int CCC(int n,int m){
    return 1ll*fac[n]*invfac[m]%mod*invfac[n-m]%mod;
}
int main(){
    n=read();a=read();b=read();c=read();d=read();
    fac[0]=1;
    FOR(i,1,n) fac[i]=1ll*fac[i-1]*i%mod;
    invfac[n]=qpow(fac[n],mod-2);
    ROF(i,n-1,0) invfac[i]=1ll*invfac[i+1]*(i+1)%mod;
    FOR(i,0,min(n/4,min(a,min(b,min(c,d))))){
        int s=1ll*CCC(n-3*i,i)*calc(i)%mod;
//      printf("s=%d\n",s);
        if(i%2==0) ans=(ans+s)%mod;
        else ans=(ans-s+mod)%mod;
    }
    printf("%d\n",ans);
}

おすすめ

転載: www.cnblogs.com/1000Suns/p/11518645.html