2023ICPC网络预选赛（1）D.Transitivity

题目传送门：

PTA

题目大意：

说实话，正式比赛的时候我是真没读懂这题啥意思到底，孩子阅读理解能力真的很差。这题还卡输入输出，真是罪恶。一个点也算是一个连通块啊！！！意思就是给你一个无向图，把不完全是连通块的补成完全是一个连通块的，然后要是给的都是连通块的，也要至少让两个连通块合成一个连通块。

题目思路：

分成两类：

第一类：就是存在不完全是连通块的，那么只需要把连通块补全就可以，一个完整的连通块含有的边数是(n-1)*n/2，减去他现有的边m，得到的就是至少需要加的边，然后遍历所有连通块，把所有需要加的边全部都加起来输出答案就可以了。

第二类：就是给你的无向图已经全部是连通块了，但是题目要求还是要至少加入一条边，那么考虑任意选择两个连通块合成一个就可以，根据贪心的思想，我们肯定要选两个含有点数最小的连通块让他们合成才能使加入的边数最小，所以，我们要记录所有的连通块的点和边数，再sort排序一下，找到两个最小的，看看他们需要加入多少条边就可以了。

代码显示如下：

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int N = 1e6 + 10;
int n, m;
int d = 1, e = 0;
typedef pair<int, int>pii;
vector<int>g[N];
bool st[N];

void dfs(int u) {
	for (auto v : g[u]) {
		e++;
		if (st[v]) continue;
		st[v] = 1;
		d++;
		dfs(v);
	}
}

void solve() {
	scanf("%lld%lld", &n, &m);
	while (m--) {
		int a, b;
		scanf("%lld%lld", &a, &b);
		g[a].push_back(b);
		g[b].push_back(a);
	}

	vector<pii> q;

	for (int i = 1; i <= n; i++) {
		if (st[i]) continue;
		st[i] = true;
		e = 0, d = 1;
		dfs(i);
		q.push_back({d, e / 2}); //注意是e/2,存的是无向图，有两条线连着
	}
	int ans = 0;
	for (auto [d, e] : q) {
		int sum = d * (d - 1) / 2;
		ans += (sum - e);

	}

	if (ans == 0) {
		sort(q.begin(), q.end());
		int sumd = q[0].first + q[1].first;
		int totd = sumd * (sumd - 1) / 2;
		ans = totd - (q[0].second + q[1].second);
	}
	printf("%lld\n", ans);





}

signed main() {
	int t = 1;
	//	cin>>t;
	while (t--) {
		solve();
	}
}