题解:
记
fi,j
f
i
,
j
为
i
i
的子树中深度为的个数,
gi,j
g
i
,
j
表示
i
i
的子树中已经匹配的对数, 第三个点到的距离需要
j
j
。
长链剖分,然后暴力扫非重儿子的子树,这样做是的。 因为每条长链只会被父节点扫一次, 且每条长链不相交。
注意delete释放内存,不然会变慢。
#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int RLEN=1<<18|1;
inline char nc() {
static char ibuf[RLEN],*ib,*ob;
(ib==ob) && (ob=(ib=ibuf)+fread(ibuf,1,RLEN,stdin));
return (ob==ib)?-1:*ib++;
}
inline int rd() {
char ch=nc(); int i=0,f=1;
while(!isdigit(ch)) {if(ch==-1) return 0; if(ch=='-')f=-1; ch=nc();}
while(isdigit(ch)) {i=(i<<1)+(i<<3)+ch-'0'; ch=nc();}
return i*f;
}
const int N=1e5+50;
int n,son[N],mxd[N];
LL ans;
vector <int> edge[N];
deque <LL> *f[N], *g[N];
inline void dfs(int x,int f) {
for(int e=edge[x].size()-1;e>=0;e--) {
int v=edge[x][e]; if(v==f) continue;
dfs(v,x);
if(mxd[v]+1>mxd[x]) mxd[x]=mxd[v]+1, son[x]=v;
}
}
inline void upt(int x,int p,LL v) {
ans+=g[x]->operator[](p)*v;
}
inline void add_f(int x,int p,LL v) {
f[x]->operator[](p)+=v;
}
inline void add_g(int x,int p,LL v) {
g[x]->operator[](p)+=v;
}
inline void dfs2(int x,int fa) {
if(son[x]) {
dfs2(son[x],x);
f[x]=f[son[x]];
g[x]=g[son[x]];
f[x]->push_front(0);
g[x]->pop_front();
g[x]->push_back(0);
g[x]->push_back(0);
}
if(f[x]==NULL) f[x]=new deque <LL>, f[x]->push_back(0);
if(g[x]==NULL) g[x]=new deque <LL>, g[x]->push_back(0);
upt(x,0,1); add_f(x,0,1);
for(int e=edge[x].size()-1;e>=0;e--) {
int v=edge[x][e]; if(v==fa || v==son[x]) continue;
dfs2(v,x);
for(int i=0;i<=mxd[v];i++)
upt(x,i+1,f[v]->operator[](i));
for(int i=1;i<=mxd[v];i++)
upt(v,i,f[x]->operator[](i-1));
for(int i=0;i<=mxd[v];i++)
add_g(x,i+1,f[x]->operator[](i+1)*f[v]->operator[](i));
for(int i=1;i<=mxd[v];i++)
add_g(x,i-1,g[v]->operator[](i));
for(int i=0;i<=mxd[v];i++)
add_f(x,i+1,f[v]->operator[](i));
delete f[v], delete g[v];
}
}
int main() {
n=rd();
for(int i=1;i<n;i++) {
int x=rd(), y=rd();
edge[x].push_back(y);
edge[y].push_back(x);
}
dfs(1,0);
dfs2(1,0);
delete f[1], delete g[1];
cout<<ans;
}