bzoj 4555 [Tjoi2016&Heoi2016]求和——NTT+第二类斯特林数

题目:https://www.lydsy.com/JudgeOnline/problem.php?id=4555

第二类斯特林数展开式:

\( S(i,j) = \frac{1}{j!} \sum\limits_{k=0}^{j}(-1)^{k}C_{j}^{k}(j-k)^{i} \)

大概是容斥枚举空的盒子个数。https://www.cnblogs.com/gzy-cjoier/p/8426987.html

在这道题里,先把 j 提到前面,再把组合数展开,推一推式子发现 j 之后的那部分是卷积的形式。就能 O( nlogn + n )了。

别把 >>1 写成 <<1 !!!!!

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int N=1e5+5,M=(1<<18)+5,mod=998244353;
int n,len,r[M],a[M],b[M],ans,jcn[N];
void upd(int &x){x>=mod?x-=mod:0;x<0?x+=mod:0;}
int pw(int x,int k)
{int ret=1;while(k){if(k&1)ret=(ll)ret*x%mod;x=(ll)x*x%mod;k>>=1;}return ret;}
void ntt(int *a,bool fx)
{
  for(int i=0;i<len;i++)
    if(i<r[i])swap(a[i],a[r[i]]);
  for(int R=2;R<=len;R<<=1)
    {
      int Wn=pw( 3,fx?(mod-1)-(mod-1)/R:(mod-1)/R );
      for(int i=0,m=R>>1;i<len;i+=R)
    for(int j=0,w=1;j<m;j++,w=(ll)w*Wn%mod)
      {
        int x=a[i+j], y=(ll)w*a[i+m+j]%mod;
        a[i+j]=x+y;  upd(a[i+j]);
        a[i+m+j]=x+mod-y;  upd(a[i+m+j]);
      }
    }
  if(!fx)return ; int inv=pw(len,mod-2);
  for(int i=0;i<len;i++)a[i]=(ll)a[i]*inv%mod;
}
int main()
{
  scanf("%d",&n);
  jcn[0]=1;for(int i=1;i<=n;i++)jcn[i]=(ll)jcn[i-1]*i%mod;
  jcn[n]=pw(jcn[n],mod-2); for(int i=n-1;i>=0;i--)jcn[i]=(ll)jcn[i+1]*(i+1)%mod;
  for(int i=0,fx=1;i<=n;i++,fx=-fx)
    {
      a[i]=fx*jcn[i]+mod;upd(a[i]);
    }
  for(int i=0;i<=n;i++)
    {
      int k=1-i;
      if(!k)
    {
      b[i]=(ll)(n+1)*jcn[i]%mod;
    }
      else
    {
      int d=pw(i,n+1);
      if(k<0)k+=mod;  k=pw(k,mod-2);
      b[i]=(ll)(1+mod-d)*k%mod*jcn[i]%mod;
    }
    }
  for(len=1;len<=n<<1;len<<=1);
  for(int i=0;i<len;i++)r[i]=(r[i>>1]>>1)+((i&1)?len>>1:0);//len>>1!! not <<1
  ntt(a,0);  ntt(b,0);
  for(int i=0;i<len;i++)a[i]=(ll)a[i]*b[i]%mod;
  ntt(a,1);
  for(int i=0,jc=1,lj=1;i<=n;i++,jc=(ll)jc*i%mod,lj<<=1,upd(lj))
    ans=(ans+(ll)lj*jc%mod*a[i])%mod;
  printf("%d\n",ans);
  return 0;
}

 

posted on 2018-12-02 17:03  Narh  阅读(206)  评论(0编辑  收藏  举报

导航