codeforce 161 D

#include <stdio.h>
#include <string.h>
#define max 50010
int n,k;
long long dp[max][511],ans;
int edge[2*max],head[2*max],next[2*max],cnt,vis[max];
void add(int v,int u)
{
    edge[cnt]=u;
    next[cnt]=head[v];
    head[v]=cnt++;
}
void init()
{
    memset(head,-1,sizeof(head));
    memset(next,-1,sizeof(next));
    memset(dp,0,sizeof(dp));
    memset(vis,0,sizeof(vis));
    cnt=0;
    ans=0;
}
void dfs(int v,int pre)
{
    //if(vis[v]==1)return;
    vis[v]=1;
    for(int i=head[v]; i!=-1; i=next[i])
    {
        if(edge[i]==pre)
           continue;
        if(!vis[edge[i]])
        {
            dfs(edge[i],v);
            for(int j = 1 ; j <= k ; ++j) ans+=dp[edge[i]][j] * dp[v][k - j];
            for(int j = 2 ; j <= k ; ++j) dp[v][j] += dp[edge[i]][j-1];
        }
        //printf("%d ",ans);
    }
}
int main ()
{
    scanf ("%d%d",&n,&k);
    k++;
    init();
    for(int i=1;i<n; i++)
    {
        int a,b;
        scanf ("%d%d",&a,&b);
        add(a,b);
        add(b,a);
    }
    for(int i=1; i<=n; i++)
        dp[i][1]=1;
        dfs(1,0);
        printf("%I64d\n",ans);
}

赤果果树形dp，因为没用LL和边开小wa无数次

树的搜索应将父节点作为参数否则易陷入死循环，一直不会这么写果断看题解了