COGS 1215. [Tyvj Aug11] 冗余电网

★ 输入文件：ugrid.in 输出文件：ugrid.out 简单对比
时间限制：1 s 内存限制：128 MB

TYVJ八月月赛提高组第2题

测试点数目：5

测试点分值：20

--内存限制：128M

--时间限制：1s

【题目描述】

北冰洋有一座孤岛，多年来一直没电。近日，令岛民们振奋的消息传来：S国的专家要为他们修建电网！！！
孤岛上共有N个村庄，发电站要建在第K个村庄中。S国的专家要在N个村庄间修建M条输电线路，但由于地理原因，M条线路无法保证每个村庄都与第K个村庄(建有发电站)直接相连，同样，也不一定能保证每个村庄都与第K个村庄间接相连(假设A与B直接相连，B与C直接相连，那么A与C间接相连)。
然而，由于S国的专家智商实在太“高”了，以至于设计出了许多冗余线路。现给出第i条线路两个端点Ui,Vi(分别表示线路连接的两个村庄,Ui!=Vi)和长度Li，请你帮岛民算一下：如果电网可以覆盖全岛，最少需要多长的电线；若不能，有多少个村庄无电可用。
注意：0<=冗余线路数目<m；部分数据有重边。电网可双向导电。

【输入格式】

第一行：N M K
接下来M行：Ui Vi Li
具体含义见题目描述

【输出格式】

如果电网可以覆盖全岛，输出最少需要的电线长度；
若不能，输出无电可用的村庄的个数。

【样例输入】

【样例输出】

【样例1】

4

【样例2】

3

【提示】

样例解释：
对于样例一，电网可以覆盖全岛，最短长度为4；
对于样例二，电网无法覆盖3,4,5这3个村庄。

数据范围：
对于20%的数据，有1<m,n<=10;
对于60%的数据，有1<m,n<=1000;
对于100%的数据，有1<m,n<=200000,0<li<=1e+7;
对于40%的数据，电网无法覆盖全岛。

【来源】

From tbcaaa8 http://www.tyvj.cn/Problem_Show.aspx?id=1591

kruskal

(rank1 蛤蛤)

屠龙宝刀点击就送

#include <algorithm>

#include <cstdio>

#define N 200005

using namespace std;

struct Edge

{

    int x,y,z;

    bool operator<(Edge a)const

    {

        return z<a.z;

    }

}e[N];

int n,m,k,cnt,fa[N];

int find_(int x) {return fa[x]==x?x:fa[x]=find_(fa[x]);}

int Main()

{

    freopen("ugrid.in","r",stdin);

    freopen("ugrid.out","w",stdout);

    scanf("%d%d%d",&n,&m,&k);

    for(int i=;i<=n;++i) fa[i]=i;

    for(int x,y,z;m--;)

    {

        scanf("%d%d%d",&x,&y,&z);

        e[++cnt]=(Edge){x,y,z};

    }

    sort(e+,e++cnt);

    int num=;

    long long sum=;

    for(int i=;i<=cnt;++i)

    {

        int fx=find_(e[i].x),fy=find_(e[i].y);

        if(fx==fy) continue;

        fa[fy]=fx;

        num++;

        sum+=(long long)e[i].z;

        if(num==n-) break;

    }

    num=,k=find_(k);

    for(int i=;i<=n;++i) if(find_(i)!=k) num++;

    if(num) printf("%d\n",num);

    else printf("%lld\n",sum);

    return ;

}

int sb=Main();

int main(int argc,char *argv[]){;}