CF1609D Social Network(并查集)

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

Problem - 1609D - Codeforcesicon-default.png?t=M4ADhttps://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;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值