Codeforces 161D

树形DP：

要求找出树上距离为k的点的对数；

对于每个节点，经过这个节点的符合条件的的点有两种：

第一种：距离他为i的儿子和他爸爸中距离他爸爸为k-i-1；(不是符合的点对中的一个)

第二种：他儿子中距离为其儿子为k-1的点；(此节点为符合条件的点对中的一个)

 

#include <cstdio>
#include <vector>
#define N 50009
using namespace std;
vector<int>vec[N];
int dp[N][505];
long long ans;
void dfs(int x,int f,int k)
{
    dp[x][0]=1;
    for(int i=0;i<vec[x].size();i++)
    {
        int q=vec[x][i];
        if(q==f)continue;
        dfs(q,x,k);
        for(int j=0;j<k;j++)
            ans+=dp[q][j]*dp[x][k-j-1];
        for(int j=1;j<k;j++)
            dp[x][j]+=dp[q][j-1];
    }
    ans+=dp[x][k];
}
int main()
{
    int n,k,x,y;
    scanf("%d%d",&n,&k);
    for(int i=0;i<n-1;i++)
    {
        scanf("%d%d",&x,&y);
        vec[x].push_back(y);
        vec[y].push_back(x);
    }
    ans=0;
    dfs(1,0,k);
    printf("%lld",ans);
    return 0;
}