bzoj 2159: Crash 的文明世界 树形dp+第二类斯特林数+排列组合

题意

给出一棵树和一个常数m，对于每一个i(1<=i<=n)，求出$\sum_{j=1}^ndist(i,j)^m$。
n<=50000,k<=150

分析

首先有一个结论，就是$x^n=\sum_{i=1}^nS(n,i)*F(x,i)$
其中$S(n,i)为第二类斯特林数，S(n,i)=S(n-1,i)*i+S(n-1,i-1)，F(x,i)表示排列，F(x,i)=\frac{x!}{(x-i)!}$
证明的话其实很简单，左边的组合意义就是把n个不同的球放到x个不同的盒子中的方案数；右边的话就相当于枚举了这n个球要放到多少个盒子里。

有了这个结论的话，题目就变成了求

$$\sum_{j=1}^n\sum_{k=1}^mS(m,k)*F(dist(i,j),k)=\sum_{k=1}^mS(m,k)\sum_{j=1}^nF(dist(i,j),k)$$
但我们发现这样还是不好求。注意到 $F(x,i)=C(x,i)/i!$那么式子就变成了 $$\sum_{k=1}^mS(m,k)*k!*\sum_{j=1}^nC(dist(i,j),k)$$
再用上pascal定理 $C(n,m)=C(n-1,m)+C(n-1,m-1)$我们就可以通过树形dp来求得后面sigma的那部分。
设down[x,i]表示当上面的k=i时x的子树对x的贡献，up[x,i]表示除了x子树外的节点对x的贡献。
设v为x的儿子，容易推出
down[x][k]=down[x][k]+down[v][k]+down[v][k-1]
up[x][k]=up[x][k]+up[fa][k]+up[fa][k-1]
up[x][k]=up[x][k]+down[fa][k]+down[fa][k-1]
up[x][k]=up[x][k]-down[x][k]-down[x][k-1]
up[x][k]=up[x][k]-down[x][k-1]-down[x][k-2]
up的最后两个递推式减去的分别是x的子树对up[fa,k]和up[fa,k-1]的贡献。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

typedef long long LL;

const int N=100005;
const int M=205;
const int MOD=10007;

int n,m,cnt,last[N],up[N][M],down[N][M],L,now,A,B,Q,s[M][M],jc[M];
struct edge{int to,next;}e[N*2];

void addedge(int u,int v)
{
    e[++cnt].to=v;e[cnt].next=last[u];last[u]=cnt;
    e[++cnt].to=u;e[cnt].next=last[v];last[v]=cnt;
}

void dfs1(int x,int fa)
{
    down[x][0]=1;
    for (int i=last[x];i;i=e[i].next)
    {
        int v=e[i].to;
        if (v==fa) continue;
        dfs1(e[i].to,x);
        down[x][0]+=down[v][0];
        for (int k=1;k<=m;k++) down[x][k]=(down[x][k]+down[v][k]+down[v][k-1])%MOD;
    }
}

void dfs2(int x,int fa)
{
    if (fa)
    {
        up[x][0]=n-down[x][0];
        for (int k=1;k<=m;k++)
        {
            up[x][k]=(up[x][k]+up[fa][k]+up[fa][k-1])%MOD;
            up[x][k]=(up[x][k]+down[fa][k]+down[fa][k-1])%MOD;
            up[x][k]=(up[x][k]-down[x][k]-down[x][k-1])%MOD;
            up[x][k]=(up[x][k]-down[x][k-1])%MOD;
            if (k>1) up[x][k]=(up[x][k]-down[x][k-2])%MOD;
            up[x][k]=(up[x][k]+MOD)%MOD;
        }
    }
    for (int i=last[x];i;i=e[i].next)
        if (e[i].to!=fa) dfs2(e[i].to,x);
}

int main()
{
    scanf("%d%d%d%d%d%d%d",&n,&m,&L,&now,&A,&B,&Q);
    for (int i=1;i<n;i++){
        now=(now*A+B)%Q;
        int tmp=i<L?i:L;
        int x=i-now%tmp,y=i+1;
        addedge(x,y);
    }
    /*scanf("%d%d",&n,&m);
    for (int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        addedge(x,y);
    }*/
    s[0][0]=jc[0]=1;
    for (int i=1;i<=m;i++)
    {
        jc[i]=jc[i-1]*i%MOD;
        for (int j=1;j<=i;j++) s[i][j]=(s[i-1][j]*j%MOD+s[i-1][j-1])%MOD;
    }
    dfs1(1,0);
    dfs2(1,0);
    for (int i=1;i<=n;i++)
    {
        int ans=0;
        for (int k=1;k<=m;k++) ans=(ans+(LL)s[m][k]*jc[k]*(up[i][k]+down[i][k])%MOD)%MOD;
        printf("%d\n",ans);
    }
    return 0;
}