CF161D 树形dp

【题意】：给出一棵树，每条边的长度为1，问树上有多少点对的距离恰好为k。

【题解】：用树的分治可解，但是写起来比较麻烦。
               设状态dp[i][j]表示以i为根的子树的节点到i的距离恰为j的个数。

               然后利用乘法原理求出答案，再把儿子的信息传给父亲。

 1 #include<cstdlib>
 2 #include<iostream>
 3 #include<cstdio>
 4 #include<cmath>
 5 #include<cstring>
 6 #include<string>
 7 #include<map>
 8 #include <algorithm>
 9 #include<set>
 #include<vector>
 #define LL long long
 #define inf 0x7fffffff
 #define E 1e-9
 #define N 50009
 #define M 509
 using namespace std;
 int m,n,k;
 LL d[N][M];
 vector<int> g[N];
 LL ans;
 void dfs(int x,int f)
 {
     int s=g[x].size();
     d[x][0]=1;//本身自己
     for(int i=0; i<s; i++)
     {
         int u=g[x][i];
         if(u==f)
             continue;
         dfs(u,x);
         for(int j=0; j<k; j++)
             ans+=d[u][j]*d[x][k-1-j];//相加等于k-1
         for(int j=1; j<=k; j++)
             d[x][j]+=d[u][j-1];
     }
 }
 int main()
 {
 #ifndef ONLINE_JUDGE
     freopen("ex.in","r",stdin);
 #endif
     scanf("%d%d",&n,&k);
     ans=0;//全局
     int x,y;
     for(int i=1; i<n; ++i)
     {
         scanf("%d%d",&x,&y);
         g[x].push_back(y);
         g[y].push_back(x);
     }
     dfs(x,-1);
     printf("%I64d\n",ans);
 
     return 0;

55 }