标签：mit vector tar ack -- swap har sig splay
\(\text{Problem}:\)第一类斯特林数·列
\(\text{Solution}:\)
与计算第二类斯特林数一列的方法类似的，设 \(F(x)\) 表示第 \(1\) 列第一类斯特林数的 \(\text{EGF}\)，有：
利用多项式快速幂可以做到 \(O(n\log n)\)。
\(\text{Code}:\)
#include <bits/stdc++.h>
#pragma GCC optimize(3)
//#define int long long
#define ri register
#define mk make_pair
#define fi first
#define se second
#define pb push_back
#define eb emplace_back
#define is insert
#define es erase
#define vi vector<int>
#define vpi vector<pair<int,int>>
using namespace std; const int N=550010, Mod=167772161;
inline int read()
{
int s=0, w=1; ri char ch=getchar();
while(ch<‘0‘||ch>‘9‘) { if(ch==‘-‘) w=-1; ch=getchar(); }
while(ch>=‘0‘&&ch<=‘9‘) s=(s<<3)+(s<<1)+(ch^48), ch=getchar();
return s*w;
}
int n,m,fac[N+5],inv[N+5];
int rev[N],r[26][2];
vector<int> a,F;
inline int ksc(int x,int p) { int res=1; for(;p;p>>=1, x=1ll*x*x%Mod) if(p&1) res=1ll*res*x%Mod; return res; }
inline void Get_Rev(int T) { for(ri int i=0;i<T;i++) rev[i]=(rev[i>>1]>>1)|((i&1)?(T>>1):0); }
inline void DFT(int T,vector<int> &s,int type)
{
for(ri int i=0;i<T;i++) if(rev[i]<i) swap(s[i],s[rev[i]]);
for(ri int i=2,cnt=1;i<=T;i<<=1,cnt++)
{
int wn=r[cnt][type];
for(ri int j=0,mid=(i>>1);j<T;j+=i)
{
for(ri int k=0,w=1;k<mid;k++,w=1ll*w*wn%Mod)
{
int x=s[j+k], y=1ll*w*s[j+mid+k]%Mod;
s[j+k]=(x+y)%Mod;
s[j+mid+k]=x-y;
if(s[j+mid+k]<0) s[j+mid+k]+=Mod;
}
}
}
if(!type) for(ri int i=0,inv=ksc(T,Mod-2);i<T;i++) s[i]=1ll*s[i]*inv%Mod;
}
inline void NTT(int n,int m,vector<int> &A,vector<int> &B)
{
int len=n+m;
int T=1;
while(T<=len) T<<=1;
Get_Rev(T);
A.resize(T), B.resize(T);
for(ri int i=n+1;i<T;i++) A[i]=0;
for(ri int i=m+1;i<T;i++) B[i]=0;
DFT(T,A,1), DFT(T,B,1);
for(ri int i=0;i<T;i++) A[i]=1ll*A[i]*B[i]%Mod;
DFT(T,A,0);
}
void GetInv(int n,vector<int> &F,vector<int> &G)
{
if(n==1) { F[0]=ksc(G[0],Mod-2); return; }
GetInv((n+1)/2,F,G);
int T=1;
while(T<=n+n) T<<=1;
Get_Rev(T);
vector<int> A,B;
A.resize(T), B.resize(T);
for(ri int i=0;i<n;i++) A[i]=F[i], B[i]=G[i];
DFT(T,A,1), DFT(T,B,1);
for(ri int i=0;i<T;i++) A[i]=(2ll*A[i]%Mod-1ll*B[i]*A[i]%Mod*A[i]%Mod+Mod)%Mod;
DFT(T,A,0);
for(ri int i=0;i<n;i++) F[i]=A[i];
}
inline void GetDao(int n,vector<int> &A,vector<int> &B)
{
for(ri int i=0;i<n-1;i++) A[i]=1ll*(i+1)*B[i+1]%Mod;
A[n-1]=0;
}
inline void GetJi(int n,vector<int> &A,vector<int> &B)
{
for(ri int i=1;i<n;i++) A[i]=1ll*B[i-1]*fac[i-1]%Mod*inv[i]%Mod;
A[0]=0;
}
inline void GetLn(int n,vector<int> &F,vector<int> &G)
{
vector<int> A,B;
A.resize(n), B.resize(n);
GetDao(n,A,G);
GetInv(n,B,G);
NTT(n,n,A,B);
GetJi(n,F,A);
}
void GetExp(int n,vector<int> &F,vector<int> &G)
{
if(n==1) { F[0]=1; return; }
GetExp((n+1)/2,F,G);
vector<int> C;
C.resize(n);
GetLn(n,C,F);
vector<int> A,B;
int T=1;
while(T<=n+n) T<<=1;
Get_Rev(T);
A.resize(T), B.resize(T);
for(ri int i=0;i<n;i++) A[i]=F[i], B[i]=(G[i]-C[i]+Mod)%Mod; B[0]++;
DFT(T,A,1), DFT(T,B,1);
for(ri int i=0;i<T;i++) A[i]=1ll*A[i]*B[i]%Mod;
DFT(T,A,0);
for(ri int i=0;i<n;i++) F[i]=A[i];
}
signed main()
{
fac[0]=1;
for(ri int i=1;i<=N;i++) fac[i]=1ll*fac[i-1]*i%Mod;
inv[N]=ksc(fac[N],Mod-2);
for(ri int i=N;i;i--) inv[i-1]=1ll*inv[i]*i%Mod;
r[25][1]=ksc(3,5), r[25][0]=ksc(ksc(3,Mod-2),5);
for(ri int i=24;~i;i--) r[i][0]=1ll*r[i+1][0]*r[i+1][0]%Mod, r[i][1]=1ll*r[i+1][1]*r[i+1][1]%Mod;
n=read(), m=read();
if(m>n)
{
for(ri int i=0;i<=n;i++) printf("0 ");
puts(""); return 0;
}
for(ri int i=0;i<m;i++) printf("0 ");
n++;
a.resize(n), F.resize(n);
for(ri int i=1;i<n;i++) a[i-1]=1ll*fac[i-1]*inv[i]%Mod;
GetLn(n,F,a);
for(ri int i=0;i<n;i++) F[i]=1ll*F[i]*m%Mod;
vector<int> Ans;
Ans.resize(n);
GetExp(n,Ans,F);
for(ri int i=m;i<n;i++) printf("%d ",1ll*Ans[i-m]*fac[i]%Mod*inv[m]%Mod);
puts("");
return 0;
}
标签：mit vector tar ack -- swap har sig splay
原文地址：https://www.cnblogs.com/zkdxl/p/14675420.html
