CodeForces 161D Distance in Tree【树形DP】

<题目链接>

题目大意：
一颗无向无环树，有n个顶点，求其中距离为k的点对数是多少，(u,v)与(v,u)为同一点对。

 #include <cstdio>
 #include <cstring>
 #include <algorithm>
 using namespace std;

 #define N int(5e4+10)
 int n,k,ans,cnt;
 int dp[N][],head[N];
 struct Edge{
     int to,nxt;
 }edge[N<<];
 void init(){
     cnt=;ans=;
     memset(head,-,sizeof(head));
     memset(dp,,sizeof(dp));
 }
 void addedge(int u,int v){
     edge[cnt].to=v,edge[cnt].nxt=head[u];
     head[u]=cnt++;
 }
 void dfs(int u,int fa){
     dp[u][]=;        //dp[i][j]表示以u为根的子树中，距u的距离为j的点的数量
     for(int i=head[u];~i;i=edge[i].nxt){
         int v=edge[i].to;
         if(v==fa)continue;
         dfs(v,u);
         for(int j=;j<k;j++)ans+=dp[u][j]*dp[v][k-j-];
         for(int j=;j<=k;j++)dp[u][j]+=dp[v][j-];
     }
 }
 int main(){
     while(~scanf("%d%d",&n,&k)){
         init();
         for(int i=;i<n;i++){
             int u,v;scanf("%d%d",&u,&v);
             addedge(u,v);addedge(v,u);
         }
         dfs(,-);
         printf("%d\n",ans);
     }
 }

2019-02-07