COGS——T 7. 通信线路

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

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

问题描述

假设要在n个城市之间建立通信联络网，则连通n个城市只需要n-1条线路。这时, 如何在最少经费的前提下建立这个通信网。在每两个城市之间都可以设置—条线路，相应地都要付出一定的经济代价。n个城市之间，最多可能设置n(n- 1)/2条线路，那么，如何在这些可能的线路中选择n-1条，以使总的耗费最少呢？

 

【输入格式】

输入文件有若干行
第一行，一个整数n，表示共有n个城市
第2--n+1行,每行n个数，分别表示该城市与其它城市之间路线的费用，如果城市间不能建立通信则用-1表示

 

【输出格式】

一行，1个整数，表示最少总费用

 

【输入输出样例】

 

输入文件

 

6 
-1 5 -1 -1 -1 -1 
5 -1 50 -1 -1 10
-1 50 -1 20 10 -1
-1 -1 20 -1 60 30
-1 -1 10 60 -1 100
-1 10 -1 30 100 -1

 

输出文件

 

75

 

【数据规模】

 

对于40％的数据，保证有n<100： 
对于60％的数据，保证有n<256； 
对于全部的数据，保证有n<=1501。

 

 #include<algorithm>
 #include<cstdio>

 using namespace std;

 const int N();
 int n,tot,fa[N];
 struct Edge
 {
     int u,v,w;
 } edge[N*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=; int if_=;char ch=getchar();
     for(; ch<''||ch>''; ch=getchar()) if(ch=='-') if_=;
     for(; ch>=''&&ch<=''; ch=getchar()) x=x*+ch-'';
     x=!if_?x:((~x)+);
 }

 int main()
 {
     freopen("mcst.in","r",stdin);
     freopen("mcst.out","w",stdout);
     read(n);
     for(int x,i=; i<=n; i++)
         for(int j=; j<=n; j++)
         {
             read(x);
             if(x!=-)
             {
                 edge[++tot].u=i,
                 edge[tot].v=j;
                 edge[tot].w=x;
             }
         }
     for(int i=; i<=n; i++) fa[i]=i;
     sort(edge+,edge++tot,cmp);
     int cnt=,ans=;
     for(int i=; i<=tot; i++)
     {
         int x=edge[i].u,y=edge[i].v;
         int fx=find(x),fy=find(y);
         if(fx==fy) continue;
         fa[fx]=fy;
         ans+=edge[i].w;
         if(++cnt==n-) break;
     }
     printf("%d",ans);
     return ;
 }