cf codeforces 1242B 0-1 MST

https://codeforces.com/contest/1242/problem/B

找图中的0权联通块,方法一暴力bfs,方法二用并查集维护,要判断是否合并,即判断是否两个集合是否存在一条0边相连,即从一个点连向那个集合的所有1权边是否小于该集合的大小。

#include<iostream>
#include<algorithm>
#include<set>
#include<queue>
#define f(i,l,r) for(i=(l);i<=(r);i++)
using namespace std;
const int MAXN=100005;
struct Node{
    int u,v;
}a[MAXN];
int n,m;
int Fa[MAXN];
int num;
int vis[MAXN];
int ans;
set<int> G[MAXN],s;
queue<int> q;
void Init()
{
    int i;
    f(i,1,n){
        Fa[i]=i;
    }
    return;
}
int Find(int x)
{
    return x==Fa[x]?x:Fa[x]=Find(Fa[x]);
}
bool Union(int x,int y)
{
    x=Find(x);y=Find(y);
    if(x==y) return false;
    Fa[x]=y;
    return true;
}
void bfs(int u)
{
    q.push(u);
    s.erase(u);
    while(!q.empty()){
        int x=q.front();
        q.pop();
        if(vis[x]) continue;
        vis[x]=1;
        set<int>::iterator it;
        for(it=s.begin();it!=s.end();){
            int v=*it;
            it++;
            if(G[x].find(v)==G[x].end()){
                q.push(v);
                s.erase(v);
            }
        }
    }
}
int main()
{
    int i,j;
    int u,v;
    cin>>n>>m;
 //   Init();
    f(i,1,n){
        s.insert(i);
    }
    f(i,1,m){
        cin>>u>>v;
        G[u].insert(v);
        G[v].insert(u);
    }
    f(i,1,n){
        if(!vis[i]){
            bfs(i);
            ans++;
        }
    }
    cout<<ans-1<<endl;
    return 0;
}
#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e5+50;
const int mod =1e9+7;
typedef long long ll;
vector<int>G[maxn];
int fa[maxn];
int sz[maxn];
map<int,int>mp;
void init()
{
	for(int i=0;i<maxn;i++) fa[i] = i;
}
int findset(int x)
{
	if(x==fa[x])  return x;
	else return fa[x] = findset(fa[x]);
} 
void unin(int x,int y)
{
	int fx = findset(x); 	int fy = findset(y);
	fa[fx] = fy;
	sz[fy] +=sz[fx]; 
}
void addedge(int u,int v)
{
	G[u].push_back(v);
	G[v].push_back(u);
}
int main()
{
	ios::sync_with_stdio(false);cin.tie(0);
	int n,m;
	cin>>n>>m;	
	init();
	for(int i=1;i<=m;i++) 
	{
		int u,v; cin>>u>>v;
		addedge(u,v);
	}
	vector<int>gp;
	for(int i=1;i<=n;i++)
	{
		sz[i] = 1;
		mp.clear();
		int tot = 0; 
		//cout<<i<<endl;
		for(int j=0;j<G[i].size();j++)
		{
			int v = G[i][j];
			if(v>=i) continue;
			int fv = findset(v);
			//cout<<v<<" -> "<<fv<<endl;
			mp[fv]++;
		}
		for(int j=0;j<gp.size();j++)
		{
			int t = findset(gp[j]);
			int q = findset(i);
			if(t==q) continue;
			if(sz[t]>mp[t])
			{
				unin(i,t);
			}
		}
		int fx = findset(i);
		if(fx==i) gp.push_back(fx);
	}
	int cnt =0;
	for(int i=1;i<=n;i++)
	{
		 if(fa[i]==i) cnt++;
	} 
	cout<<cnt-1<<endl;
	return 0;
}
发布了100 篇原创文章 · 获赞 9 · 访问量 2万+

猜你喜欢

转载自blog.csdn.net/MrTinTin/article/details/103299482