EOJ 3497/EOJ Monthly 2018.2 D.黑心的出租车

题目简介

---

给定一个树/森林，从1号点出发，遍历森林中所有的点最后回到1号点。
大巴可以在有边相连的两点间往返，而出租车可以在任意两点间往返。
要求：乘出租车次数最少的情况下，乘大巴次数也最少，求两个次数。

说明

---

官方题解：

因为题目保证无环，因此必为树或森林。
当图为一棵树时，答案为 0,2(n−1)。
当图为森林时，出租车次数判一下联通块即可，其实答案就是 n−m。而巴士站数，对于每个联通块，需要 2 倍边的数量减去树的直径，然后求和即可。
注意慎用 memset，初始化仅需 DFS 所需要用的点，不然会 TLE。

#include <bits/stdc++.h>
using namespace std;

const int maxn = 1e5+5;
int dep[maxn];
vector<int> g[maxn];
bool vis[maxn];

inline void dfs(int x, int& root, int y=0)
{
    dep[x] = dep[y]+1;
    vis[x] = 1;
    if (dep[x] > dep[root]) root = x;
    for (auto i: g[x])
    {
        if (i == y) continue;
        dfs(i, root, x);
    }
}

int main()
{
    int n, k, u, v, root1, root2;
    cin >> n >> k;
    for (int i = 1; i <= k; ++i)
    {
        scanf("%d%d", &u, &v);
        g[u].push_back(v);
        g[v].push_back(u);
    }
    int cnt = k<<1;
    if (k == n-1) printf("0 %d\n", cnt);
    else
    {
        printf("%d ", n-k);
        for (int i = 1; i <= n; ++i)
            if (!vis[i])
            {
                root1 = root2 = 0;
                dfs(i, root1); dfs(root1, root2);
                cnt -= dep[root2]-1;
            }
        printf("%d\n", cnt);
    }
    return 0;
}