codeforces161D. Distance in Tree(树形DP)

思路：

dp[i][j]表示点 i 的子节点到点 i 的距离为j的路径个数。设一点为u，其子节点为v， u、v间的边权值为w

转移方程为：dp[u][j]+=dp[v][j-1];

每个点对答案的贡献为： ans+=dp[u][j]*dp[v][(k-1-j)];

#include <iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#define ll long long
using namespace std;
vector <int>mp[50005];
vector <int>w[50005];
int dp[50005][505],k,n;
ll ans=0;
void dfs(int fa,int now){
    dp[now][0]=1;
    for(int i=0;i<mp[now].size();i++){
        int to=mp[now][i];
        if(to==fa)continue;
        dfs(now, to);
        for(int j=0;j<k;j++){
                ans+=dp[now][j]*dp[to][(k-1-j)];
        }
        for(int j=1;j<=k;j++){
                dp[now][j]+=dp[to][j-1];
        }
    }
}
int main(){
        cin>>n>>k;
        for(int i=0;i<=n;i++){
            mp[i].clear();
            w[i].clear();
            for(int j=0;j<=k;j++)
                dp[i][j]=0;
        }
        for(int i=1;i<n;i++){
            int a,b;
            cin>>a>>b;
            mp[a].push_back(b);
            mp[b].push_back(a);
        }
        ans=0;
        dfs(0, 1);
        cout<<ans<<endl;
    return 0;
}