[无根树] Keeping On Track Gym101673F

题意

有编号从0到n的n+1个城市和n条路径，每两个城市之间有且只有一条路径。当两个城市间存在一天可达路径时，称两个城市是联通的。将关键城市定义为，当去掉这个城市后不连通的城市对最多。求去掉关键城市后不连通的城市对有多少，当新添加一条路径后，仍不连通城市对的最小数目。

题解

从题目的描述中可以将城市看成一颗无根树。因此在最开始，我很自然的想到了求这棵无根树的重心。然而，本题的答案并不是树的重心。不加证明就做题，只凭自己的猜测，也难怪自己WA到死。。。。

自己之前没有接触过无根树，但求重心的方法提供了解决的思路。

排bug排不出来，还是要考虑做法的正确性啊！

AC代码

#include <cstdio>
#include <queue>
#include <cstring>
#include <vector>
#include <functional>
using namespace std;

typedef long long ll;

const int maxn = 100100;
const int INF = 0x3f3f3f3f;

ll d[maxn];
vector<int> G[maxn];
int n;
ll ans, m1, m2;

ll dfs(int u, int fa){
    d[u] = 1;
    ll bm1=0, bm2=0, temp = 0;
    for(int i = 0; i<G[u].size(); i++)if(G[u][i] != fa){
        ll b = dfs(G[u][i], u);
        temp += (n-b)*b;
        d[u] += b;
        if(b>=bm1){ bm2 = bm1; bm1 = b;}
        else if(b>bm2) bm2 = b;
    }
    if(n+1-d[u]>=bm1){ bm2 = bm1; bm1 = n+1-d[u]; }
    else if(n+1-d[u]>bm2) bm2 = n+1-d[u];
    temp += (d[u]-1)*(n+1-d[u]);
    temp /= 2;
    if(temp>ans){
        ans = temp;
        m1 = bm1;
        m2 = bm2;
    }
    return d[u];
}

int main(){
    while(scanf("%d", &n)!= EOF){
        for(int i = 0; i<=maxn; i++) G[i].clear();
        memset(d, 0, sizeof(d));
        int u,v;
        for(int i = 0; i<n; i++){
            scanf("%d%d", &u, &v);
            G[u].push_back(v);
            G[v].push_back(u);
        }
        ans = -1; m1 = m2 = 0;
        dfs(0, -1);
        printf("%lld %lld\n", ans, ans-m1*m2);
    }
}