CF1609D Social Network（并查集）

最新推荐文章于 2024-11-10 19:25:12 发布

eva_can(not)survive

最新推荐文章于 2024-11-10 19:25:12 发布

阅读量194

点赞数

分类专栏：图论文章标签：图论算法 c++

本文链接：https://blog.csdn.net/Ghostttttttiii/article/details/124872725

版权

图论专栏收录该内容

43 篇文章 0 订阅

订阅专栏

本文主要介绍了使用并查集与优先队列的数据结构来解决一类问题：通过合并操作计算某个状态下的集合元素个数。在每轮合并后，利用优先队列计算当前最大值以及次大值，用于输出结果。代码中涉及到了并查集的路径压缩和按秩合并优化，以及优先队列的使用技巧。

摘要由CSDN通过智能技术生成

Problem - 1609D - Codeforceshttps://codeforces.com/problemset/problem/1609/D

int pre[MAXN];
int n, k;
bool vis[MAXN];
int sum[MAXN];

inline void init() {
	for (int i = 1; i <= n; i++) {
		pre[i] = i;
		sum[i] = 1;
	}
}

int find(int x) {
	if (x == pre[x])
		return x;
	return pre[x] = find(pre[x]);
}

inline bool same(int x, int y) {
	return find(x) == find(y);
}


int main() {
	int x, y;
	scanf("%d %d", &n, &k);
	init();
	int cnt = 0;
	priority_queue<int> que;
//	int mx=0;
	for (int i = 1; i <= k; i++) {
//		scanf("%d %d",&s[i].x,&s[i].y);
		scanf("%d %d", &x, &y);
//		vis[x]=1,vis[y]=1;
		if (same(x, y)) {
			cnt++;
		} else {
			x = find(x);
			y = find(y);
			pre[y] = x;
			sum[x] += sum[y];
		}
		while (!que.empty()) {
			que.pop();
//			printf("===\n");
		}
		for (int i = 1; i <= n; i++) {
			if (i == find(i)) {
				que.push(sum[i]);
			}
		}
		int ans = 0;
		ans += que.top();
		que.pop();
		for (int i = 1; i <= cnt; i++) {
			ans += que.top();
			que.pop();
		}
		printf("%d\n", ans - 1);
	}
	return 0;
}