2023ICPC网络赛第一场 D Transitivity

题意：给一张n个点m条边的简单无向图，对于任意的两个顶点 u，v 如果它们之间存在路径，那么他们就必须有一条边相连，你可以往无向图里加边（至少加一条），问最少需要加多少条边。

思路：如果两个点本来是不连通的，那么一定不会加边使他们连通，因为这样还要加很多边，使得这个新的连通分量满足要求。所以将所有连通分量分开考虑， 对于一个有 d 个顶点， e 条边的连通分量，我需要加上 d * (d - 1) / 2 - e 条边使得这个连通分量满足要求，将所有连通分量需要加的边的数量累积在一起就是答案。如果所有连通分量本身就满足要求，但是题目要求至少加一条边，此时必须将两个点的数量最少得连通分量连接在一起。

#include<bits/stdc++.h>
using namespace std;
#define pb push_back
const int N = 1e6 + 10;
typedef long long ll;
typedef pair<ll, ll> PII;
vector<int> g[N];
bool vis[N];
ll e = 0, d = 0;
void dfs(int u) {
    for (int v : g[u]) {
        e++;
        if (vis[v]) continue;
        vis[v] = 1;
        d++;
        dfs(v);
    }
}
void solve() {
    int n, m;
    cin >> n >> m;
    for (int i = 0; i < m; i++) {
        int u, v;
        cin >> u >> v;
        g[u].pb(v), g[v].pb(u);
    }
    vector<PII> p;
    for (int i = 1; i <= n; i++) {
        if (vis[i]) continue;
        vis[i] = 1;
        e = 0, d = 1;
        dfs(i);
        p.pb({d, e / 2});
    }
    ll ans = 0;
    for (auto [d, e] : p) {
        ll sum = d * (d - 1) / 2;
        ans += sum - e;
    }
    if (!ans) {
        sort(p.begin(), p.end());
        ll d = p[0].first + p[1].first;
        ll sum = d * (d - 1) / 2;
        ans = sum - (p[0].second + p[1].second);
    }
    cout << ans;
}
int main() {
    ios::sync_with_stdio(false), cin.tie(nullptr), cout.tie(nullptr);
    int tt = 1;
    //cin >> tt;
    while (tt--) {
        solve();
    }
    return 0;
}