hrbust 1359 Leyni的国家II （树形dp）

Leyni的国家II

Time Limit: 2000 MS Memory Limit: 262144 K Total Submit: 34(12 users) Total Accepted: 17(11 users) Rating:  Special Judge: No

Description

Leyni经过了若干年的征战，终于建立了自己的国家，这个国家包含n个城市，编号为1到n，而且这些城市中，任意两个城市间有且只有一种路线（这个路线可能需要经过若干条城市之间的公路），每条城市之间的公路的长度都相等（都是1单位长度）。 Leyni现在想知道有多少对城市之间的最短路径长度等于k。（城市u到城市v，城市v到城市u算一种）

Input

本题有多组测试数据，输入的第一行是一个整数T代表着测试数据的数量，接下来是T组测试数据。 对于每组测试数据： 第1行 包含两个以空格分隔的整数n和k (1 ≤ n ≤ 50000, 1 ≤ k ≤ 500)。 第2 .. n行 在这n – 1行中，每行包含两个以空格分隔的整数u和v (1 ≤ u, v ≤ n, u ≠ v)，表示城市u和城市v之间存在一条双向公路。（数据保证所有给出的边都是不同的）

Output

对于每组测试数据： 第1行 输出有多少对城市之间的最短路径长度等于k。

Sample Input

1 5 2 1 2 2 3 3 4 2 5

Sample Output

4

Hint

第一组样例的中最短路径长度等于2的路径有(1, 3) (1, 5) (3, 5) (2, 4)。

Author

齐达拉图@HRBUST

 

解题思路：对于任意相距为k的节点有两种情况

1.其中一个点为父节点另一个点位子节点 这个时候就是dp[son[i]][k-1];

2。两个点都是子节点，把父节点当作中转点，相距为k的情况。这个时候要枚举子节点。

#include<iostream>
#include<cstdio>
#include<vector>
#include<queue>
#include<set>
#include<algorithm>
#include<cstring>
using namespace std;
#define rep(i,j,k) for(int i=j;i<=k;i++)
#define sca(x) scanf("%d",&x)
#define per(i,j,k) for(int i=j;i>=k;i--)
#define inf 0x3f3f3f3f
#define LL long long
#define N 1600
#define inf 0x3f3f3f3f

int dp[50005][505];
int vis[50005];
vector<int>v[50005];
LL ans=0;
int k;

void dfs(int now)
{
    int son[10005],top=0;
    vis[now]=1;
    dp[now][0]=1;
    for(int i=0;i<v[now].size();i++)
    {
        int to=v[now][i];
        if(vis[to])continue;
        dfs(to);
        son[top++]=to;//记录 子节点
        for(int i=1;i<=k;i++)dp[now][i]+=dp[to][i-1];//父节点的状态由子节点转过来
        ans+=dp[to][k-1]; //这里是第一种情况
    }
    rep(i,0,top-1)//处理第二种情况
    {
        rep(j,i+1,top-1)
        {
            rep(l,0,k-2)
            {
                if(k-2-l>=0)
                {
                    ans+=dp[son[i]][l]*dp[son[j]][k-2-l];//把父节点当作中转点。
                }
            }
        }
    }
}
int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        int n;
        ans=0;
        memset(vis,0,sizeof(vis));
        memset(dp,0,sizeof(dp));
        scanf("%d%d",&n,&k);
        rep(i,1,n)v[i].clear();
        rep(i,2,n)
        {
            int x,y;
            scanf("%d%d",&x,&y);
            v[x].push_back(y);
            v[y].push_back(x);
        }
        dfs(1);
        printf("%lld\n",ans);
    }
}
/*
1
7 2
1 2
1 3
3 7
2 4
2 5
2 6
*/