COGS——T 1215. [Tyvj Aug11] 冗余电网

http://www.cogs.pro/cogs/problem/problem.php?pid=1215

★ 输入文件：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

并查集判断是否在同一个连通块、最小生成树找ans

 #include <algorithm>

 #include <cstdio>

 using namespace std;

 const int N();

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

 struct Edge

 {

     int u,v,w;

 }road[N<<];

 bool cmp(Edge a,Edge b)

 {

     return a.w<b.w;

 }

 int find(int x)

 {

     return fa[x]==x?x:fa[x]=find(fa[x]);

 }

 inline void read(int &x)

 {

     x=; register char ch=getchar();

     for(;ch>''||ch<'';) ch=getchar();

     for(;ch>=''&&ch<='';ch=getchar()) x=x*+ch-'';

 }

 int AC()

 {

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

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

     read(n),read(m),read(k);

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

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

     {

         read(road[i].u),

         read(road[i].v),

         read(road[i].w);

     }

     long long ans=; int cnt=;

     sort(road+,road+m+,cmp);

     for(int fx,fy,i=;i<=m;i++)

     {

         fx=find(road[i].u),fy=find(road[i].v);

         if(fa[fy]==fx) continue;

         fa[fy]=fx;

         ans+=(long long)road[i].w;

         if(++cnt==n-) break;

     }

     cnt=;k=find(k);

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

         if(find(i)!=k) cnt++;

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

     else  printf("%lld",ans);

     return ;

 }

 int I_want_AC=AC();

 int main(){;}