BZOJ 5093: [Lydsy1711月赛]图的价值

第二类斯特林数模版题

需要一些组合数的小$ trick$

upd：这里更新了本题巧妙的$ O(k)$做法，虽然常数很大就是了

---

传送门：here

题意：求所有$ n$个节点的无重边自环图的价值和，定义一张图的价值为每个点度数的$ k$次方和，点有标号

---

$ Solution$

显然每个节点的贡献是独立的

枚举每个节点的度数，和这个点不联通的边可连可不连

$ ans=n*2^{\frac{(n-1)(n-2)}{2}}\ \ \sum\limits_{i=0}^{n-1}i^kC_{n-1}^i$

我们实际要求解的东西就是$ f(n,m)=\sum\limits_{i=0}^ni^mC_n^i$

把$i^m$用斯特林数展开得
$f(n,m)=\sum\limits_{i=0}^n\sum\limits_{j=0}^mC_i^jS(m,j)j!C_n^i$

把$j$移动到前面得
$f(n,m)=\sum\limits_{j=0}^mS(m,j)j!\sum\limits_{i=0}^nC_i^jC_n^i$
考虑后面这个$\sum\limits_{i=0}^nC_i^jC_n^i$是什么
本质相当于在$n$个物品中选出集合$A$，再在集合$A$中选取$j$个物品
也就是在$n$个物品中选取$j$个物品，其他$n-j$个物品可在集合$A$中也可不在
因此$\sum\limits_{i=0}^nC_i^jC_n^i=C_n^j2^{n-j}$
$f(n,m)=\sum\limits_{j=0}^mS(m,j)j!C_n^j2^{n-j}$
$NTT$筛出斯特林数直接计算即可
复杂度$O(k \ log \ k)$

---

$my \ code$

#include<ctime>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#define p 998244353
#define file(x)freopen(x".in","r",stdin);freopen(x".out","w",stdout)
#define rt register int
#define ll long long
using namespace std;
inline ll read(){
    ll x = ; char zf = ; char ch = getchar();
    while (ch != '-' && !isdigit(ch)) ch = getchar();
    if (ch == '-') zf = -, ch = getchar();
    while (isdigit(ch)) x = x *  + ch - '', ch = getchar(); return x * zf;
}
void write(ll y){if(y<)putchar('-'),y=-y;if(y>)write(y/);putchar(y%+);}
void writeln(const ll y){write(y);putchar('\n');}
int i,j,k,m,n,x,y,z,cnt;
int a[],b[],R[],lim;
ll ksm(ll x,ll y){
    if(!y)return ;ll ew=;
    while(y>){
        if(y&)y--,ew=x*ew%p;
        y>>=,x=x*x%p;
    }return x*ew%p;
}
int inv[],S[];
struct poly{
    int n,m,lim;
    void init(int k){
        //a[i]=(-1)^i / i!    b[i] = i^k/i!
        n=k;
        a[]=;b[]=;
        for(rt i=;i<=k;i++){
            a[i]=-1ll*a[i-]*inv[i]%p;
            b[i]=ksm(i,k)*a[i]%p;
            if(i&)b[i]=-b[i];
        }
        lim=;while(lim<=n+n)lim<<=;
        for(rt i=;i<lim;i++)R[i]=(R[i>>]>>)|(i&?(lim>>):);
    }
    void NTT(int *A,int fla){
        for(rt i=;i<lim;i++)if(i<R[i])swap(A[i],A[R[i]]);
        for(rt i=;i<lim;i<<=){
            ll w=ksm(,p//i);
            for(rt j=;j<lim;j+=i<<){
                ll K=;
                for(rt k=;k<i;k++,K=K*w%p){
                    ll x=A[j+k],y=K*A[i+j+k];
                    A[j+k]=(x+y)%p;A[i+j+k]=(x-y)%p;
                }
            }
        }
        if(fla==-){
            reverse(A+,A+lim);
            for(rt i=;i<=n;i++)S[i]=1ll*A[i]*ksm(lim,p-)%p;
        }

    }
    void main(int k){
        init(k);
        NTT(a,);NTT(b,);
        for(rt i=;i<lim;i++)a[i]=1ll*a[i]*b[i]%p;
        NTT(a,-);
    }
}NTT;
int main(){
    n=read()-;k=read();
    inv[]=inv[]=;
    for(rt i=;i<=k+;i++)inv[i]=1ll*inv[p%i]*(p-p/i)%p;

    NTT.main(k);
    ll jc=,C=,ans=,sum=ksm(,n-j);
    for(rt j=;j<=k;j++){
        (ans+=S[j]*jc%p*C%p*sum)%=p;
        jc=jc*(j+)%p;C=C*(n-j)%p*inv[j+]%p;
        sum=sum*inv[]%p;
    }
    cout<<(ans*(n+)%p*ksm(,(ll)n*(n-)/)%p+p)%p;
    return ;
}