BZOJ 3522|POI 2014|Hotel|树形DP

给出树上三点对的个数，满足两两距离相等。

显然，符合题意的情况只有3个点在某个点的不同子树中的情况，因此枚举根跑一次树形DP处理出不同深度的点的个数搞一下就可以了。

2016年1月A的题为什么没有写题解呢。。
为Claris的加强版做准备

#include <cstdio>
#include <cstring>
const int N = 5001, M = 10000;
int h[N], v[M], p[M], cnt = 0;
long long f[N], g[N], depth[N];
void add(int x, int y) {
    p[++cnt] = h[x]; v[cnt] = y; h[x] = cnt;
}
void dfs(int x, int fa, int dep) {
    depth[dep]++;
    for (int i = h[x]; i; i = p[i])
        if (v[i] != fa)
            dfs(v[i], x, dep + 1);
}

int main() {
    int n, i, x, y;
    long long ans = 0;
    scanf("%d", &n);
    for (i = 1; i < n; i++) {
        scanf("%d%d", &x, &y);
        add(x, y); add(y, x);
    }
    for (x = 1; x <= n; x++) {
        memset(f, 0, sizeof f);
        memset(g, 0, sizeof g);
        for (i = h[x]; i; i = p[i]) {
            memset(depth, 0, sizeof depth);
            dfs(v[i], x, 1);
            for (y = 1; y <= n; y++) {
                ans += g[y] * depth[y];
                g[y] += f[y] * depth[y];
                f[y] += depth[y];
            }
        }
    }
    printf("%lld", ans);
    return 0;
}

3522: [Poi2014]Hotel

Description

有一个树形结构的宾馆，n个房间，n-1条无向边，每条边的长度相同，任意两个房间可以相互到达。吉丽要给他的三个妹子各开（一个）房（间）。三个妹子住的房间要互不相同（否则要打起来了），为了让吉丽满意，你需要让三个房间两两距离相同。
有多少种方案能让吉丽满意？

Input

第一行一个数n。
接下来n-1行，每行两个数x,y，表示x和y之间有一条边相连。

Output

让吉丽满意的方案数。

Sample Input

7
1 2
5 7
2 5
2 3
5 6
4 5

Sample Output

5

HINT

【样例解释】

{1,3,5},{2,4,6},{2,4,7},{2,6,7},{4,6,7}

【数据范围】

$n≤5000$