D. Distance in Tree

题目链接：http://codeforces.com/contest/161/problem/D

题意：求所有距离为k的点对(u,v) 的数量

思路：dp[u][i] 代表以u为根的子树 距离为i 的数量，  主要是考虑如何不重复的计算

可以先把每一棵子树的先记录 用乘法，然后再把贡献放到u上，每次保证通过子树的某一条边

来做到不重不漏

#include<bits/stdc++.h>
using namespace std;
const int maxn=5e4+10;
const int mod=1e9+7;
#define ll long long
#define ull unsigned long long
#define pi pair<int,int>
#define fi first
#define sc second
#define pb push_back
#define ull unsigned long long
ll ans;
ll dp[maxn][505];
int n,k;
vector<int>E[maxn];


void dfs(int u,int fa)
{
    dp[u][0]=1;
    for(auto &v:E[u])
    {
        if(v==fa) continue;
        dfs(v,u);
        for(int i=0;i<k;i++)
            ans+=dp[v][k-1-i]*dp[u][i];
        for(int i=1;i<=k;i++)
            dp[u][i]+=dp[v][i-1];
    }
}

int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    cin>>n>>k;
    for(int i=1;i<n;i++)
    {
        int x,y;
        cin>>x>>y;
        E[x].pb(y);
        E[y].pb(x);
    }
    dfs(1,0);
    cout<<ans<<'\n';




}