传送门
解析:
长链剖分裸题。
我们考虑如果树上三个点距离相等,那么必然存在一个点 u u u,它到这三个点的距离全部相等。
那么分类讨论两种情况,三个点在 u u u节点的不同子树内部,这个很好统计,随手就转移了。
那么还有一个情况,其中一个节点在子树外部。
换句话说,就是在
u
u
u的某一个祖先的其他子树当中。
这个不好直接在 u u u处统计啊。
换一个思路,我们考虑在这个上级祖先处统计这个三元组。
我们需要对节点 u u u记录数组 g u g_u gu,其中 g u , i g_{u,i} gu,i表示需要向子树外延伸距离为 i i i来寻找第三个点的二元组。
这个东西每条长链可以 O ( d e p ) O(dep) O(dep)处理,总的复杂度 O ( n ) O(n) O(n)。
然后就是长链剖分的套路了,指针申请空间。其中 g g g数组需要开两倍空间,因为转移方向是 − 1 -1 −1,所以向上一倍合并给父亲,向下一倍合并其他子树
代码:
#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define re register
#define gc get_char
#define pc put_char
#define cs const
namespace IO{
namespace IOONLY{
cs int Rlen=1<<16|1;
char buf[Rlen],*p1,*p2;
char obuf[Rlen],*p3=obuf;
char ch[23];
}
inline char get_char(){
using namespace IOONLY;
return (p1==p2)&&(p2=(p1=buf)+fread(buf,1,Rlen,stdin),p1==p2)?EOF:*p1++;
}
inline void put_char(char c){
using namespace IOONLY;
*p3++=c;
if(p3==obuf+Rlen)fwrite(obuf,1,Rlen,stdout),p3=obuf;
}
inline void FLUSH(){
using namespace IOONLY;
fwrite(obuf,1,p3-obuf,stdout),p3=obuf;
}
inline int getint(){
re int num;
re char c;
while(!isdigit(c=gc()));num=c^48;
while(isdigit(c=gc()))num=(num+(num<<2)<<1)+(c^48);
return num;
}
inline void outint(ll a){
using namespace IOONLY;
if(a==0)pc('0');
while(a)ch[++ch[0]]=a-a/10*10,a/=10;
while(ch[0])pc(ch[ch[0]--]^48);
}
}
using namespace IO;
cs int N=100004;
vector<int> edge[N];
inline void addedge(int u,int v){
edge[u].push_back(v);
edge[v].push_back(u);
}
int n,fcnt,gcnt;ll ans;
ll *f[N],*g[N],farr[N],garr[N<<1];
int mxdep[N],son[N];
inline void dfs1(int u,int fa){
mxdep[u]=1;
for(int re e=0;e<edge[u].size();++e){
re int v=edge[u][e];
if(v==fa)continue;
dfs1(v,u);
if(mxdep[v]>mxdep[son[u]])son[u]=v;
mxdep[u]=max(mxdep[u],mxdep[v]+1);
}
}
inline void dfs2(int u,int fa){
++f[u][0];
if(!son[u])return ;
else{
re int v=son[u];
f[v]=f[u]+1;
g[v]=g[u]-1;
dfs2(v,u);
ans+=f[u][0]*g[u][0];
}
for(int re e=0;e<edge[u].size();++e){
re int v=edge[u][e];
if(v==fa||v==son[u])continue;
f[v]=farr+fcnt;fcnt+=mxdep[v];
gcnt+=mxdep[v];g[v]=garr+gcnt;gcnt+=mxdep[v];
dfs2(v,u);
for(int re i=0;i+1<mxdep[v];++i)ans+=f[u][i]*g[v][i+1];
for(int re i=1;i-1<mxdep[v];++i)ans+=g[u][i]*f[v][i-1];
for(int re i=0;i-1<mxdep[v];++i){
g[u][i]+=g[v][i+1];
if(i>=1)g[u][i]+=f[v][i-1]*f[u][i];
}
for(int re i=1;i-1<mxdep[v];++i)f[u][i]+=f[v][i-1];
}
}
signed main(){
n=getint();
for(int re i=1;i<n;++i)addedge(getint(),getint());
dfs1(1,0);
f[1]=farr+fcnt;fcnt+=mxdep[1];
gcnt+=mxdep[1];g[1]=garr+gcnt;gcnt+=mxdep[1];
dfs2(1,0);
outint(ans);pc('\n');
FLUSH();
return 0;
}