[第二类斯特林数 FFT] BZOJ5093. 图的价值_第二类斯特林数任意模数 fft-CSDN博客

本文链接：https://blog.csdn.net/Coldef/article/details/78762556

怕是一道套路题…然而这种题还是不太会
考虑每个点的贡献都是相同的
那么答案就是

n 2 (n - 1 2) \sum i = 0 n - 1 (n - 1 i) i k

$n2^{n-1\choose 2}\sum_{i=0}^{n-1}{n-1\choose i}i^k$

$i^k$ 用第二类斯特林数带进去得到

n 2 (n - 1 2) \sum i = 0 n - 1 (n - 1 i) \sum j = 0 k S (k, j) j! (i j)

$n2^{n-1\choose 2}\sum_{i=0}^{n-1}{n-1\choose i}\sum_{j=0}^kS(k,j)j!{i\choose j}$

n 2 (n - 1 2) \sum j = 0 k S (k, j) j! \sum i = 1 n - 1 (n - 1 i) (i j)

$n2^{n-1\choose 2}\sum_{j=0}^{k}S(k,j)j!\sum_{i=1}^{n-1}{n-1\choose i}{i\choose j}$

而 $\sum_{i=1}^{n-1}{n-1\choose i}{i\choose j}$ 等于 $\sum_{j=0}^k{n-1\choose j}2^{n-1-j}$

所以得到

a n s = n 2 (n - 1 2) \sum j = 0 k S (k, j) j! (n - 1 j) 2 n - 1 - j

$ans=n2^{n-1\choose 2}\sum_{j=0}^{k}S(k,j)j!{n-1\choose j}2^{n-1-j}$

第二类斯特林数满足 $S(n,k)={1\over k!}\sum_{j=0}^k(-1)^{k-j}{k\choose j}j^n$ ，所以可以用生成函数来搞

组合数用 ${n\choose m+1}={n-m\over m+1}{n\choose m}$ 递推一下

#include <cstdio>
#include <iostream>
#include <algorithm>

using namespace std;

typedef long long ll;

const int N=600010,P=998244353;

int num,L,n,k,a[N],S[N],fac[N],inv[N],rev[N],C[N];

inline int Pow(int x,ll y){
  int ret=1;
  for(;y;y>>=1,x=1LL*x*x%P) if(y&1) ret=1LL*ret*x%P;
  return ret;
}

int w[2][N];

void Pre(const int &n){
  int g=Pow(3,(P-1)/num);
  w[0][0]=w[1][0]=1;
  for(int i=1;i<n;i++) w[1][i]=1LL*w[1][i-1]*g%P;
  for(int i=1;i<n;i++) w[0][i]=w[1][n-i];

  fac[0]=1; for(int i=1;i<=num;i++) fac[i]=1LL*fac[i-1]*i%P;
  inv[1]=1; for(int i=2;i<=num;i++) inv[i]=1LL*(P-P/i)*inv[P%i]%P;
  inv[0]=1; for(int i=1;i<=num;i++) inv[i]=1LL*inv[i-1]*inv[i]%P;
}

inline void NTT(int *a,int n,int r){
  for(int i=0;i<n;i++) if(rev[i]>i) swap(a[i],a[rev[i]]);
  for(int i=1;i<n;i<<=1)
    for(int j=0;j<n;j+=i<<1)
      for(int k=0;k<i;k++){
    int x=a[j+k],y=1LL*w[r][n/(i<<1)*k]*a[j+k+i]%P;
    a[j+k]=(x+y)%P; a[j+k+i]=(x+P-y)%P;
      }
  if(!r) for(int i=0,iv=Pow(n,P-2);i<n;i++) a[i]=1LL*a[i]*iv%P;
}

int main(){
  freopen("1.in","r",stdin);
  freopen("1.out","w",stdout);
  scanf("%d%d",&n,&k);
  for(num=1;num<=k;num<<=1,L++); num<<=1;
  for(int i=1;i<num;i++) rev[i]=(rev[i>>1]>>1)|((i&1)<<L);
  Pre(num);
  for(int i=0,j=1;i<=k;i++,j=-j)
    a[i]=(P+j*inv[i])%P,S[i]=1LL*Pow(i,k)*inv[i]%P;
  NTT(a,num,1); NTT(S,num,1);
  for(int i=0;i<num;i++) S[i]=1LL*S[i]*a[i]%P;
  NTT(S,num,0);
  int ans=0;
  C[0]=1;
  for(int i=0;i<k;i++)
    C[i+1]=1LL*C[i]*(n-i-1)%P*Pow(i+1,P-2)%P;
  /*int cur=0;
  for(int i=0;i<=5;i++)
    cur=(cur+1LL*S[i]*fac[i]%P*C[i])%P;
    cout<<Pow(6,5)<<' '<<cur<<endl;*/
  for(int i=0;i<=k;i++)
    ans=(ans+1LL*S[i]*fac[i]%P*C[i]%P*Pow(2,n-1-i))%P;
  ans=1LL*ans*n%P*Pow(2,1LL*(n-1)*(n-2)/2)%P;
  printf("%d\n",ans);
  return 0;
}