【題解】Luogu P5339 [TJOI2019]唱、跳、rap和籃球

原題傳送門

這題zsy寫的是\(O(n^2)\),還有NTT\(O(n^2\log n)\)的作法。個人是暴力,\(O(\frac{a b n}{4})\),足夠經過

考慮設\(f(i)\)表示序列中至少有\(i\)組人討論cxk的方案數

這樣就珂以進行容斥,易知答案ans爲:

\[ans=\sum_{i=0}^{Min(n/4,a,b,c,d)} (-1)^i f(i)\]

咱們考慮如何計算\(f(i)\)

若是視討論cxk的組爲一個元素,則一共有\(n-3*i\)個元素

咱們把問題轉換成一個多重排列的方案數

多重排列的方案數求法:

如今有\(m\)個不一樣的元素,每一個\(i\)元素有\(a_i\)個,那麼方案數爲

\[(\sum_{i=1}^m a_i)! \times \prod_{i=1}^m \frac{1}{a_i!}\]

那麼咱們只要暴力計算便可

#include <bits/stdc++.h>
#define ll long long 
#define N 2005
#define mod 998244353
#define getchar nc
using namespace std;
inline char nc(){
    static char buf[100000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline int read()
{
    register int x=0,f=1;register char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+ch-'0',ch=getchar();
    return x*f;
}
inline void write(register int x)
{
    if(!x)putchar('0');if(x<0)x=-x,putchar('-');
    static int sta[20];register int tot=0;
    while(x)sta[tot++]=x%10,x/=10;
    while(tot)putchar(sta[--tot]+48);
}
inline int Min(register int a,register int b)
{
    return a<b?a:b;
}
int n,a,b,c,d,lim;
ll fac[N],inv[N],f[N],res,ans; 
int main()
{
    n=read(),a=read(),b=read(),c=read(),d=read();
    fac[0]=fac[1]=inv[0]=inv[1]=1;
    for(register int i=2;i<=n;++i)
        fac[i]=fac[i-1]*i%mod;
    for(register int i=2;i<=n;++i)
        inv[i]=(mod-mod/i)*inv[mod%i]%mod;
    for(register int i=2;i<=n;++i)
        inv[i]=inv[i]*inv[i-1]%mod;
    lim=Min(n>>2,Min(Min(a,b),Min(c,d)));
    for(register int x=0,v;x<=lim;++x)
    {
        v=(x&1)?-1:1;
        for(register int i=0;i<=n;++i)
            f[i]=0;
        for(register int i=0;i<=a-x;++i)
            for(register int j=0;j<=Min(n-4*x-i,b-x);++j)
                f[i+j]=(f[i+j]+inv[i]*inv[j])%mod;
        res=0;
        for(register int i=0;i<=c-x;++i)
            for(register int j=0;j<=Min(n-4*x-i,d-x);++j)
                res=(res+inv[i]*inv[j]%mod*f[n-4*x-i-j])%mod;
        res=res*fac[n-3*x]%mod*inv[x]%mod;
        ans=(ans+v*res)%mod;
    }
    write(ans<0?ans+mod:ans);
    return 0;
}
相關文章
相關標籤/搜索