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;
}