【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=0,f=1;
    char ch=getchar();
    while(!isdigit(ch)){
        if(ch=='-')    f=-1;
        ch=getchar();
    }
    while(isdigit(ch)){
        num=num*10+ch-'0';
        ch=getchar();
    }
    return num*f;
}

struct Edge{
    int next,to;
}edge[maxn*2];
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=1;i<n;++i){
        int x=read(),y=read();
        add(x,y);
        add(y,x);
        d[x]++;    d[y]++;
    }
    for(int i=1;i<=n;++i)    f[i][0]=read();
    for(int i=1;i<=n;++i){
        f[i][1]+=f[i][0];
        for(int j=head[i];j;j=edge[j].next){
            int to=edge[j].to;
            f[i][1]+=f[to][0];
        }
    }
    for(int j=2;j<=m;++j)
        for(int i=1;i<=n;++i){
            for(int k=head[i];k;k=edge[k].next){
                int to=edge[k].to;
                f[i][j]+=f[to][j-1];
            }
            f[i][j]-=f[i][j-2]*(d[i]-1);
        }
    for(int i=1;i<=n;++i)    printf("%lld
",f[i][m]);
    return 0;
}