【Luogu】P3047附近的牛（树形DP）

　　题目链接

　　树形DP，设f[i][j]是当前在i点，j步之内有多少牛。从相邻点to的f[to][j-1]转移而来，减去重复计算即可。

　　

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cctype>
#include<cstdlib>
#define maxn 150020
#define maxk 22
using namespace std;
inline long long read(){
    long long num=,f=;
    char ch=getchar();
    while(!isdigit(ch)){
        if(ch=='-')    f=-;
        ch=getchar();
    }
    while(isdigit(ch)){
        num=num*+ch-'';
        ch=getchar();
    }
    return num*f;
}

struct Edge{
    int next,to;
}edge[maxn*];
int head[maxn],num;
inline void add(int from,int to){
    edge[++num]=(Edge){head[from],to};
    head[from]=num;
}

long long f[maxn][maxk];
long long d[maxn];

int main(){
    int n=read(),m=read();
    for(int i=;i<n;++i){
        int x=read(),y=read();
        add(x,y);
        add(y,x);
        d[x]++;    d[y]++;
    }
    for(int i=;i<=n;++i)    f[i][]=read();
    for(int i=;i<=n;++i){
        f[i][]+=f[i][];
        for(int j=head[i];j;j=edge[j].next){
            int to=edge[j].to;
            f[i][]+=f[to][];
        }
    }
    for(int j=;j<=m;++j)
        for(int i=;i<=n;++i){
            for(int k=head[i];k;k=edge[k].next){
                int to=edge[k].to;
                f[i][j]+=f[to][j-];
            }
            f[i][j]-=f[i][j-]*(d[i]-);
        }
    for(int i=;i<=n;++i)    printf("%lld\n",f[i][m]);
    return ;
}