CF1060E Sergey and Subway 题解

题目链接
我的洛谷Blog

两百年前看题解做过的一道题，自己还写了个题解，现在回来看感觉当时的做法极其不严谨。

设 $\mathrm{dis}(i,j)$ 表示 $i,j$ 的距离。
我们可以发现，对于每一个点对 $(i,j)$：

• 如果原来 $\mathrm{dis}(i,j)$ 为偶数，那么加上新边后就变成了 $\frac{\mathrm{dis}(i,j)}{2}$。
• 如果原来 $\mathrm{dis}(i,j)$ 为奇数，那么加边后就变成了 $\frac{\mathrm{dis}(i,j)+1}{2}$。

那么，最终的答案就是 ∑ 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$ 为 $x$ 的子树大小，$d_x$ 为 $x$ 的深度；$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)​。

总时间复杂度 $\mathcal{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;
}