两百年前看题解做过的一道题,自己还写了个题解,现在回来看感觉当时的做法极其不严谨。
设
dis
(
i
,
j
)
\operatorname{dis}(i,j)
dis(i,j) 表示
i
,
j
i,j
i,j 的距离。
我们可以发现,对于每一个点对
(
i
,
j
)
(i,j)
(i,j):
- 如果原来 dis ( i , j ) \operatorname{dis}(i,j) dis(i,j) 为偶数,那么加上新边后就变成了 dis ( i , j ) 2 \frac{\operatorname{dis}(i,j)}{2} 2dis(i,j)。
- 如果原来 dis ( i , j ) \operatorname{dis}(i,j) dis(i,j) 为奇数,那么加边后就变成了 dis ( i , j ) + 1 2 \frac{\operatorname{dis}(i,j)+1}{2} 2dis(i,j)+1。
那么,最终的答案就是 ∑ 1 ≤ i ≤ j ≤ n { dis ( i , j ) + ( dis ( i , j ) m o d 2 ) } 2 \frac{\sum\limits_{1 \le i \le j \le n} \{ \operatorname{dis}(i,j)+(\operatorname{dis}(i,j) \mod 2) \}}{2} 21≤i≤j≤n∑{dis(i,j)+(dis(i,j)mod2)}。
设 s x s_x sx 为 x x x 的子树大小, d x d_x dx 为 x x x 的深度; u , v u,v u,v 分别为深度为奇数和偶数的点个数。
那么答案就可以化简为 ∑ 1 ≤ i ≤ j ≤ n { s x ⋅ ( n − s x ) } + ( u × v ) 2 \frac{\sum\limits_{1 \le i \le j \le n}\{s_x\cdot(n-s_x)\}+(u\times v)}{2} 21≤i≤j≤n∑{sx⋅(n−sx)}+(u×v)。
总时间复杂度 O ( n ) \mathcal O(n) O(n)。
为了方便理解,本代码未开 long long
。
#include<cstdio>
#include<iostream>
#include<vector>
using namespace std;
const int Maxn=200000+20,inf=0x3f3f3f3f;
vector <int> e[Maxn];
int s[Maxn],d[Maxn];
int n,cnt,ans;
inline int read()
{
int s=0,w=1;
char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
while(ch>='0' && ch<='9')s=(s<<3)+(s<<1)+(ch^48),ch=getchar();
return s*w;
}
void dfs(int x,int fa)
{
s[x]=1,d[x]=d[fa]+1;
for(int i=0;i<e[x].size();++i)
{
int y=e[x][i];
if(y==fa)continue;
dfs(y,x);
s[x]+=s[y];
}
}
int main()
{
//freopen("in.txt","r",stdin);
n=read();
for(int i=1;i<n;++i)
{
int x=read(),y=read();
e[x].push_back(y);
e[y].push_back(x);
}
dfs(1,0);
for(int i=1;i<=n;++i)
{
ans+=(s[i]*(n-s[i]));
if(d[i] & 1)cnt++;
}
ans+=(cnt*(n-cnt));
ans>>=1;
printf("%lld\n",ans);
return 0;
}