并查集求联通块个数【洛谷P1197】

传送门:https://www.luogu.org/problemnew/show/P1197

并查集求联通块个数。

这个题是每摧毁一个点求一次联通块的个数,并查集不容易维护这种摧毁更新的,相反的,并查集可以很好的维护每安装一个点更新一次这种问题。摧毁和安装是对立的,既然按顺序摧毁,我们就逆序安装。

首先要算出没安装前(也就是摧毁之后)的联通块的个数。怎么求呢,我们在记录图的边的时候记录一下起点和终点,然后我们for一遍边集,如果起点和终点都没有被摧毁,我们就将他们放进并查集。

然后首先计算一下最终答案。并查集每一个find(i)==i都是一个联通块。

然后我们开始安装点。

每安装一个点,先将tot++,然后遍历它的边,如果它的相邻点没有被摧毁(或者建好了),也就是相邻点还在,并且他们不是一个集合,我们就让tot--,并且将该点放进并查集。(为什么要tot--,大家自己动手画个简单的图就能明白了。)

大致思路就是这样啦!

下面是大家喜闻乐见的代码(话说我觉得自己的码风还不错哎)

#include <bits/stdc++.h>
using namespace std;
typedef pair<int,int> P;
const int maxn = 5e5+7;
int p[maxn],r[maxn];
int b[maxn];
int cur[maxn];
int ans[maxn];
vector<int> G[maxn];
vector<pair<int,int> >e;
void init()
{
	for(int i=0;i<maxn;i++)
	{
		p[i] = i;
		r[i] = 0;
	}
}
int find(int x)
{
	if(x==p[x]) return x;
	return p[x] = find(p[x]);
}
void unite(int x,int y)
{
	x = find(x);
	y = find(y);
	if(x==y) return;
	if(r[x]<r[y])
	{
		p[x] = y;
	}
	else
	{
		p[y] = x;
		if(r[x]==r[y])
		{
			r[x]++;
		}
	}
}
bool same(int x,int y)
{
	return find(x)==find(y);
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	init();
	int n,m;
	cin>>n>>m;
	for(int i=0;i<m;i++)
	{
		int x,y;
		cin>>x>>y;
		G[x].push_back(y);
		G[y].push_back(x);
		e.push_back(P(x,y));
	}
	int k;
	cin>>k;
	for(int i=0;i<k;i++)
	{
		cin>>b[i];
		cur[b[i]] = 1;
	}
	for(int i=0;i<e.size();i++)
	{
		int x = e[i].first;
		int y = e[i].second;
		if(!cur[x] && !cur[y])
		{
			unite(x,y);
		}
	}
	int tot = 0;
	for(int i=0;i<n;i++)
	{
		if(find(i)==i && !cur[i])
		{
			tot++;
		}
	}
	for(int i=k-1;i>=0;i--)
	{
		cur[b[i]] = 0;
		ans[i] = tot;
		tot++;
		if(G[b[i]].size())
		{
			for(int j=0;j<G[b[i]].size();j++)
			{
				if(!cur[G[b[i]][j]])
				{
					if(!same(b[i],G[b[i]][j]))
					{
						tot--;
						unite(b[i],G[b[i]][j]);
					}
				}
			}
		}
	}
	cout<<tot<<endl;
	for(int i=0;i<k;i++)
	{
		cout<<ans[i]<<endl;
	}
	return 0;
}

猜你喜欢

转载自blog.csdn.net/KIKO_caoyue/article/details/84073688
今日推荐