【题解】数字组合(NTT+组合 滑稽)

【题解】数字组合(NTT+组合 滑稽)

今天实践一下谢总讲的宰牛刀233,滑稽。

\((1+x)(1+x)(1+x)\)的\(x^2\)系数就代表了有三个一快钱硬币构成的两块钱的方案数量。

很好理解,毕竟拆括号这种东西本身就有组合意义。

那么假设面值\(i\)有\(a_i\)个,那么最终的答案是
\[ G(x)=\prod_{i=1}^{1000} (1+{a_i\choose 1}x+{a_i\choose 2}x^2\dots) \]

的\(x^m\)项系数
直接NTT即可。

//@winlere
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;  typedef long long ll;
inline int qr(){
      register int ret=0,f=0;
      register char c=getchar();
      while(c<48||c>57)f|=c==45,c=getchar();
      while(c>=48&&c<=57) ret=ret*10+c-48,c=getchar();
      return f?-ret:ret;
}
int la,lb;
const int maxn=1<<17|1;
int a[maxn],b[maxn];
const int mod=998244353;
namespace poly{

      
      inline int Pow(int base,const int&p){
        register int ret=1;
        for(register int t=p;t;t>>=1,base=1ll*base*base%mod)
          if(t&1) ret=1ll*ret*base%mod;
        return ret;
      }
      
      const int g=3;
      const int gi=Pow(g,mod-2);
      
      int A[maxn],B[maxn],r[maxn];
      int savlen=-1;
      inline void getr(const int&len){
        if(len==savlen)return;
        savlen=len;
        int cnt=0;
        for(register int t=1;t<len;t=t<<1)++cnt;
        for(register int t=0;t<len;++t)
          r[t]=r[t>>1]>>1|(t&1)<<cnt>>1;
      }
      
      inline int getlen(const int&len){
        register int ret;
        for(ret=1;ret<len;ret<<=1);
        return ret;
      }
      
      inline void NTT(int*a,const int&len,const int&tag){
        getr(len);
        for(register int t=0;t<len;++t)
          if(r[t]>t)swap(a[t],a[r[t]]);
        int *a0,*a1,s;
        if(s=g,tag!=1)s=gi;
        for(register int t=1,wn;t<len;t<<=1){
          wn=Pow(s,(mod-1)/(t<<1));
          for(register int i=0;i<len;i+=t<<1){
            a1=(a0=a+i)+t;
            for(register int k=0,w=1,temp;k<t;++k,++a0,++a1,w=1ll*w*wn%mod){
                  temp=1ll**a1*w%mod;
                  *a1=(*a0-temp)%mod;
                  *a0=(*a0+temp)%mod;
                  if(*a1<0)*a1+=mod;
            }
          }
        }
        if(tag!=1) for(register int t=0,inv=Pow(len,mod-2);t<len;++t)
                 a[t]=1ll*a[t]*inv%mod;
      }
      
      void inv(int*a,int*b,const int&len){
        if(len==1){b[0]=Pow(a[0],mod-2);return;}
        inv(a,b,len>>1);
        for(register int t=0;t<len;++t) A[t]=a[t],B[t]=b[t];
        NTT(A,len<<1,1);NTT(B,len<<1,1);
        for(register int t=0,ed=len<<1;t<ed;++t) A[t]=1ll*A[t]*B[t]%mod*1ll*B[t]%mod;
        NTT(A,len<<1,-1);
        for(register int t=0;t<len;++t) b[t]=((b[t]+b[t])%mod+mod-A[t])%mod;
      }
      
      void INV(int*a,int*b,const int&len){
        inv(a,b,getlen(len));
      }
      
      inline void MLP(int*a,int*b,int*c,const int&len1,const int&len2){
        int k=getlen(len1+len2+2);
        NTT(a,k,1);NTT(b,k,1);
        for(register int t=0;t<=k;++t) c[t]=1ll*a[t]*b[t]%mod;
        NTT(c,k,-1);
      }
}

ll jc[501],v[501];
int buk[1001];

using namespace poly;
inline ll c(const int&n,const int&m){
      if(n<m)return 0;
      return jc[n]*v[m]%mod*v[n-m]%mod;
}

int main(){
      jc[0]=v[0]=1;
      for(register int t=1;t<=500;++t){
        jc[t]=jc[t-1]*t%mod;
        v[t]=Pow(jc[t],mod-2);
      }
      int n=qr(),m=qr();
      a[0]=1;
      for(register int t=1;t<=n;++t) ++buk[qr()];
      for(register int t=1;t<=1000;++t){
        if(!buk[t])continue;
        if(t>m)break;
        memset(b,0,sizeof b);
        b[0]=1;
        for(register int i=1;1ll*t*i<=m&&i<=buk[t];++i)
          b[i*t]=c(buk[t],i),lb=i*t;
        MLP(a,b,a,la,lb);
        la+=lb;
        if(la>m+1){
          for(register int t=m+1;t<=la;++t)
            a[t]=0;
          la=m;
        }
      }
      cout<<a[m]<<endl;
      return 0;
}

【题解】数字组合(NTT+组合 滑稽)

可还行...

上一篇:【题解】BZOJ5093图的价值(二项式+NTT)


下一篇:NTT练习