题解P3047 [USACO12FEB]附近的牛Nearby Cows

题目：P3047 [USACO12FEB]附近的牛Nearby Cows

本题的意思就是求离每个点距离不超过k的点的权值和。

然后我们很容易想到，f[x][j]=Σf[v][j-1](v∈son[x])

然后我们会发现有重复的，于是容斥原理运用一下。

因为每次v扩展一步到x，到x距离为j-2的点肯定有一部分是到v距离为j-1的点，就会被重复计算。

大概这样，然后每次f[x][j]-=f[v][j-2]*(size[v]-1)。

为啥-1呢，因为你还要保留原来那一份。

但是记录这题的原因是，我发现树形dp可以不写dfs！

本来是写了dfs，感觉没法调，然后就看了题解，才知道树形dp也可以不在dfs里面进行！

因为我们这次不需要每次更新扩大size[x]，我们只关心和他相连的点怎么转移过来没所以我们不用dfs进行dp。

一个小细节就是数组多开点防止越界。

 

#include<cstdio>
#define it register int
#define il inline
const int N=1000005;
int h[N],nxt[N],adj[N],s[N],u,v,t,n,k; 
long long f[N/5][21];
il void add(){
    nxt[++t]=h[u],h[u]=t,adj[t]=v,++s[u];
    nxt[++t]=h[v],h[v]=t,adj[t]=u,++s[v];
}
il void fr(int &num){
    num=0;char c=getchar();int p=1;
    while(c<'0'||c>'9') c=='-'?p=-1,c=getchar():c=getchar();
    while(c>='0'&&c<='9') num=num*10+c-'0',c=getchar();
    num*=p;
}
il void lfr(long long &num){
    num=0;char c=getchar();int p=1;
    while(c<'0'||c>'9') c=='-'?p=-1,c=getchar():c=getchar();
    while(c>='0'&&c<='9') num=num*10+c-'0',c=getchar();
    num*=p;
}
int main(){ 
    fr(n),fr(k);
    for(it i=1;i<n;++i) fr(u),fr(v),add();
    for(it i=1;i<=n;++i) lfr(f[i][0]);
    for(it j=1;j<=k;++j)
        for(u=1;u<=n;++u){
            for(it i=h[u];i;i=nxt[i])
                f[u][j]+=f[adj[i]][j-1];
            j>1?f[u][j]-=1ll*(s[u]-1)*f[u][j-2]:f[u][1]+=f[u][0];
        }
    for(it i=1;i<=n;++i)
        printf("%lld\n",f[i][k]);
    return 0;
}