[bzoj3522][bzoj4543][POI2014]HOTEL

Description

给出一棵树，求这棵树中有多少种方法选择三个点使得这三个点之间的距离两两相等。
n<=1e5

Solution

首先我们来考虑一种奇怪的可优化的N^2做法。
设Fi,j表示i的子树中到i的距离为j的点的个数
Gi,j表示i的子树中有多少个点对满足到他们的lca的距离为d，同时lca到i的距离为d-j
也就是还能再伸出j的长度的点对数。
首先我们考虑选择一个儿子继承可以发现转移的形式相当于数组位移一位，此时的答案就是Gx,0
然后我们枚举其他儿子转移，转移的复杂度就是这个儿子往下的深度。
那么我们可以用一个。。。。听说叫做长链剖分的做法。。。
普通的重链剖分是选择size最大的儿子作为偏爱儿子
那么长链剖分我们就选择往下的深度最大的儿子作为偏爱儿子
那么每次Dp我们继承偏爱儿子的dp值，考虑数组位移是O(1)的
接着把所有的轻儿子暴力合并，可以发现复杂度相当于那个儿子所在的链的长度。
那么总复杂度就是O(n)的
至于空间的话你给每个长链分配正比于它的链长的空间就也是O(n)的
由于懒得考虑细节和加上懒得分析实际的空间就抄了栋栋的标魔改了一发
感觉指针这个东西还是很玄妙的，所以就把它改成了不用指针的
这个应该会好懂一点吧。。。

Code

#include <cstdio>
#include <cstring>
#include <algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define rep(i,a) for(int i=last[a];i;i=next[i])
using namespace std;
typedef long long ll;
const int N=1e5+5;
int n,x,y,t[N*2],next[N*2],last[N],dep[N],mx[N],l,f[N],g[N],now=5;
ll fty[N*5],gty[N*5],ans;
void add(int x,int y) {t[++l]=y;next[l]=last[x];last[x]=l;}
void dfs(int x,int y) {
    dep[x]=dep[y]+1;mx[x]=x;
    rep(i,x) 
        if (t[i]!=y) {
            dfs(t[i],x);
            if (dep[mx[t[i]]]>dep[mx[x]]) mx[x]=mx[t[i]];
        }
    rep(i,x)
        if (t[i]!=y&&(mx[x]!=mx[t[i]]||x==1)) {
            now+=dep[mx[t[i]]]-dep[x]+1;
            f[mx[t[i]]]=now;
            g[mx[t[i]]]=(now+=1);
            now+=(dep[mx[t[i]]]-dep[x])*2+1;
        }
}
void dp(int x,int y) {
    rep(i,x) 
        if (t[i]!=y) {
            dp(t[i],x);
            if (mx[t[i]]==mx[x]) {
                f[x]=f[t[i]]-1;
                g[x]=g[t[i]]+1;
            }
        }
    fty[f[x]]=1;
    ans+=gty[g[x]];
    rep(i,x)
        if (t[i]!=y&&mx[t[i]]!=mx[x]) {
            fo(j,0,dep[mx[t[i]]]-dep[x]) {
                ans+=fty[f[x]+j-1]*gty[g[t[i]]+j];
                ans+=gty[g[x]+j+1]*fty[f[t[i]]+j];
            }
            fo(j,0,dep[mx[t[i]]]-dep[x]) {
                fty[f[x]+j]+=fty[f[t[i]]+j-1];
                gty[g[x]+j]+=gty[g[t[i]]+j+1];
                gty[g[x]+j+1]+=fty[f[x]+j+1]*fty[f[t[i]]+j];
            }
        }
}
int main() {
    freopen("tree.in","r",stdin);
    freopen("tree.out","w",stdout);
    scanf("%d",&n);
    fo(i,1,n-1) {
        scanf("%d%d",&x,&y);
        add(x,y);add(y,x);
    }
    dfs(1,0);dp(1,0);
    printf("%lld\n",ans);
}