【CF1243D&CF920E】0-1 MST（bfs，set）

题意：给定一张n个点的完全图，其中有m条边权为1其余为0，求最小生成树的权值和

n，m<=1e5

思路：答案即为边权为0的边连接的联通块个数-1

用set存图和一个未被选取的点的集合，bfs过程中如果找到边权为0且未被选取的边则加入

如果要维护联通块大小也在bfs里随便记一下就好

具体实现看代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned int uint;
typedef unsigned long long ull;
typedef long double ld;
typedef pair<int,int> PII;
typedef pair<ll,ll> Pll;
typedef vector<int> VI;
typedef vector<PII> VII;
typedef pair<ll,ll>P;
#define N  500010
#define M  1000000
#define INF 1e9
#define fi first
#define se second
#define MP make_pair
#define pb push_back
#define pi acos(-1)
#define mem(a,b) memset(a,b,sizeof(a))
#define rep(i,a,b) for(int i=(int)a;i<=(int)b;i++)
#define per(i,a,b) for(int i=(int)a;i>=(int)b;i--)
#define lowbit(x) x&(-x)
#define Rand (rand()*(1<<16)+rand())
#define id(x) ((x)<=B?(x):m-n/(x)+1)
#define ls p<<1
#define rs p<<1|1
#define fors(i) for(auto i:e[x]) if(i!=p)

const int MOD=1e9+7,inv2=(MOD+1)/2;
      double eps=1e-6;
      int dx[4]={-1,1,0,0};
      int dy[4]={0,0,-1,1};

set<int>G[N],S;
int vis[N],c[N],ans;

int read()
{
   int v=0,f=1;
   char c=getchar();
   while(c<48||57<c) {if(c=='-') f=-1; c=getchar();}
   while(48<=c&&c<=57) v=(v<<3)+v+v+c-48,c=getchar();
   return v*f;
}

ll readll()
{
   ll v=0,f=1;
   char c=getchar();
   while(c<48||57<c) {if(c=='-') f=-1; c=getchar();}
   while(48<=c&&c<=57) v=(v<<3)+v+v+c-48,c=getchar();
   return v*f;
}

void bfs(int st)
{
    queue<int> q;
    S.erase(st);
    q.push(st);
    int s=1;
    while(q.size()>0)
    {
        int u=q.front();
        q.pop();
        if(vis[u]) continue;
        vis[u]=1;

        for(auto it=S.begin();it!=S.end();)
        {
            int v=*it;
            it++;
            if(G[u].find(v)==G[u].end())
            {
                q.push(v);
                S.erase(v);
                s++;
            }
        }
    }
    c[ans]=s;
}

int main()
{
    int n=read(),m=read();
    rep(i,1,n) S.insert(i);
    rep(i,1,m)
    {
        int x=read(),y=read();
        G[x].insert(y);
        G[y].insert(x);
    }
    ans=0;
    rep(i,1,n)
     if(!vis[i])
     {
         ans++;
         bfs(i);
     }
    printf("%d
",ans);
    sort(c+1,c+ans+1);
    rep(i,1,ans) printf("%d ",c[i]);
    return 0;
}