Nyoj 布线问题(并查集&&图论)

描述南阳理工学院要进行用电线路改造，现在校长要求设计师设计出一种布线方式，该布线方式需要满足以下条件：
1、把所有的楼都供上电。
2、所用电线花费最少

 

输入

第一行是一个整数n表示有n组测试数据。(n<5)
每组测试数据的第一行是两个整数v,e.
v表示学校里楼的总个数(v<=500)
随后的e行里，每行有三个整数a,b,c表示a与b之间如果建铺设线路花费为c(c<=100)。（哪两栋楼间如果没有指明花费，则表示这两栋楼直接连通需要费用太大或者不可能连通）
随后的1行里，有v个整数,其中第i个数表示从第i号楼接线到外界供电设施所需要的费用。( 0<e<v*(v-1)/2 )
（楼的编号从1开始），由于安全问题，只能选择一个楼连接到外界供电设备。
数据保证至少存在一种方案满足要求。

输出

每组测试数据输出一个正整数,表示铺设满足校长要求的线路的最小花费。

样例输入

1
4 6
1 2 10
2 3 10
3 1 10
1 4 1
2 4 1
3 4 1
1 3 5 6

样例输出

4

来源

[张云聪]原创

上传者

张云聪

自己定义的排序函数没有写在前边  Runtime Error多次......

排序之后,用并查集即可,,,Prim也可以处理....

并查集代码:

 #include <vector>
 #include <map>
 #include <set>
 #include <algorithm>
 #include <iostream>
 #include <cstdio>
 #include <cmath>
 #include <cstdlib>
 #include <string>
 #include <cstring>
 #include <queue>
 using namespace std;
 #define INF 0x3f3f3f3f
 #define MAX 1000000 

 int v;
 int dis[];
 struct node{
     int a,b,cost;
 }c[];

 bool cmp(node x,node y)
 {
     return x.cost<y.cost;
 }

 void init()
 {
     for(int i=; i<=v; i++){
         dis[i]=i;
     }
 }

 int find(int x)
 {
     if(dis[x]!=x)
         dis[x]=find(dis[x]);
     return dis[x];
 }

 int main()
 {
     int n,e;
     scanf("%d",&n);
     while(n--){
         int min=INF,m,sum=;
         scanf("%d%d",&v,&e);
         init();
         for(int i=; i<e; i++){
             scanf("%d%d%d",&c[i].a,&c[i].b,&c[i].cost);
         }
         for(int i=; i<v; i++){
             scanf("%d",&m);
             if(min>m)
                 min=m;
         }
         sort(c,c+e,cmp);
         for(int i=; i<e; i++){
             int x=find(c[i].a);
             int y=find(c[i].b);
             if(x!=y){
                 dis[x]=y;
                 sum+=c[i].cost;
             }
         }
         printf("%d\n",sum+min);
     }
     return ;
 }

Prim代码:

 #include <vector>
 #include <map>
 #include <set>
 #include <algorithm>
 #include <iostream>
 #include <cstdio>
 #include <cmath>
 #include <cstdlib>
 #include <string>
 #include <cstring>
 #include <queue>
 using namespace std;
 #define INF 0x3f3f3f3f
 #define MAX 1000000 

 int dis[],vis[],mp[][];
 int v;

 int Prim()
 {
     fill(dis,dis+v,INF);
     memset(vis,,sizeof(vis));
     int n=,ans=;
     for(int i=; i<=v; i++){
         if(i!=n)
             dis[i]=mp[n][i];
     }
     vis[]=;
     while(true){
         int min=INF;
         for(int j=; j<=v; j++){
             if(!vis[j] && min>dis[j]){
                 min=dis[j];
                 n=j;
             }
         }
         if(min==INF)
             break;
         ans+=min;
         vis[n]=;
         for(int i=; i<=v; i++){
             if(!vis[i] && dis[i]>mp[n][i])
                 dis[i]=mp[n][i];
         }
     }
     return ans;
 }

 int main()
 {
     int n,e,i,x,y,cost;
     scanf("%d",&n);
     while(n--){
         int sum=,min=INF,m;
         memset(mp,INF,sizeof(mp));
         scanf("%d%d",&v,&e);
         for(i=; i<e; i++){
             scanf("%d%d%d",&x,&y,&cost);
             mp[x][y]=mp[y][x]=cost;
         }
         for(i=; i<v; i++){
             scanf("%d",&m);
             if(min>m)
                 min=m;
         }
         sum=min+Prim();
         printf("%d\n",sum);
     }
 }