bzoj3522 [Poi2014]Hotel

Description

---

有一棵树，每条边的长度相同，求选三个点两两间距离相等的方案数

n≤5000

Solution

---

一开始全往点分治上想惹忽略了三个点的限制（一脸弱相，姿势上来了还得被吊锤

首先这三个点一定不会在一条链上，这个可以画图理解
那么可以枚举根枚举不同的子树，用t1[x]记录选一个深度为x的点的方案数，用t2[x]记录选两个深度为x的点且两个点属于不同子树的方案数，用t[x]记录在现在枚举的这棵子树内深度为x的点数量

那么分类瞎jb搞搞就出来了，注意开LL

Code

---

#include <stdio.h>
#include <string.h>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
#define fill(x,t) memset(x,t,sizeof(x))

typedef long long LL;
const int N=20005;
const int E=40005;

struct edge {int y,next;} e[E];

int t[N],t1[N],t2[N];
int ls[N],edCnt;
LL ans;

bool vis[N];

void add_edge(int x,int y) {
    e[++edCnt]=(edge) {y,ls[x]}; ls[x]=edCnt;
    e[++edCnt]=(edge) {x,ls[y]}; ls[y]=edCnt;
}

void dfs(int fa,int x,int dep) {
    t[dep]++;
    for (int i=ls[x];i;i=e[i].next) {
        if (e[i].y==fa) continue;
        dfs(x,e[i].y,dep+1);
    }
}

int main(void) {
    int n; scanf("%d",&n);
    rep(i,2,n) {
        int x,y; scanf("%d%d",&x,&y);
        add_edge(x,y);
    }
    LL ans=0;
    rep(x,1,n) {
        fill(t1,0);
        fill(t2,0);
        for (int i=ls[x];i;i=e[i].next) {
            dfs(x,e[i].y,1);
            rep(j,1,n) {
                if (!t[j]) continue;
                ans+=(LL)t2[j]*t[j];
                t2[j]+=(LL)t1[j]*t[j];
                t1[j]+=(LL)t[j];
                t[j]=0;
            }
        }
    }
    printf("%lld\n", ans);
    return 0;
}