dp+dfs好题,状态转移写得很赞……dp[i][j中的i表示以第几个点为中心,j长度,综合起来表示以i为中心长度为j的边的总数……
#include<iostream>
#include<vector>
using namespace std;
int k;
__int64 ans,dp[50005][505];
vector<int> v[50005];
void dfs(int x,int y)
{
int i,j,z;
dp[x][0]=1;
for(i=0;i<v[x].size();i++)
{
z=v[x][i];
if(y!=z)
{
dfs(z,x);
for(j=1;j<=k;j++)ans+=dp[x][j-1]*dp[z][k-j];
for(j=1;j<=k;j++)dp[x][j]+=dp[z][j-1];
}
}
}
int main()
{
//freopen("a.txt","r",stdin);
int i,n,x,y;
scanf("%d%d",&n,&k);
for(i=1;i<=n;i++)v[i].clear();
for(i=1;i<n;i++)
{
scanf("%d%d",&x,&y);
v[x].push_back(y);
v[y].push_back(x);
}
memset(dp,0,sizeof(dp));
ans=0;
dfs(1,0);
printf("%I64d\n",ans);
return 0;
}