题目vj链接

题面:

题意:
给定一张n个点的完全图,边权只有0/1两种。
给定完全图中边权为1的边,求一棵最小生成树。

题解:
由于边权为1的边的数量较少,我们可以考虑怎么利用边权为1的边。
因为我们要求最小生成树,那么肯定是要优先选取边权为0的边。
我们把所有能用边权为0的边的连通块搜出来,假设最终有 c n t cnt cnt 个连通块,这些连通块内部均可以用边权为0的边联通,那么最终最小生成树的权值为 c n t 1 cnt-1 cnt1

代码:

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<string>
#include<queue>
#include<bitset>
#include<map>
#include<unordered_map>
#include<unordered_set>
#include<set>
#define ui unsigned int
#define ll long long
#define llu unsigned ll
#define ld long double
#define pr make_pair
#define pb push_back
#define lc (cnt<<1)
#define rc (cnt<<1|1)
#define len(x) (t[(x)].r-t[(x)].l+1)
#define tmid ((l+r)>>1)
#define fhead(x) for(int i=head[(x)];i;i=nt[i])
#define max(x,y) ((x)>(y)?(x):(y))
#define min(x,y) ((x)>(y)?(y):(x))
using namespace std;

const int inf=0x3f3f3f3f;
const ll lnf=0x3f3f3f3f3f3f3f3f;
const double dnf=1e18;
const int mod=1e9+7;
const double eps=1e-8;
const double pi=acos(-1.0);
const int hp=13331;
const int maxn=100100;
const int maxm=100100;
const int maxp=100100;
const int up=30;

int head[maxn],ver[maxn<<1],nt[maxn<<1],tot=1;
int cnt=0;
unordered_set<int>ans;//当前还不在连通块中的点;

void add(int x,int y)
{
    ver[++tot]=y,nt[tot]=head[x],head[x]=tot;
}

void dfs(int x)
{
    unordered_set<int>se;
    se.clear();
    for(int i=head[x];i;i=nt[i])
    {
        auto it=ans.find(ver[i]);
        if(it!=ans.end())
        {
            se.insert(*it);
            ans.erase(*it);
        }
    }
    swap(ans,se);
    for(auto x:se)
        dfs(x);
}

int main(void)
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
        ans.insert(i);

    int x,y;
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d",&x,&y);
        add(x,y);
        add(y,x);
    }

    for(int i=1;i<=n;i++)
    {
        if(ans.find(i)!=ans.end())
        {
            ans.erase(i);
            cnt++;
            dfs(i);
        }
    }
    printf("%d\n",cnt-1);
    return 0;
}