UVALive 6437 Power Plant（最小生成树）

来源：互联网发布：常见的nosql数据库编辑：程序博客网时间：2024/06/06 20:50

UVALive 6437 Power Plant （题目链接）

题意：给定n个点，其中有k个发电站，m条有权边。要求所有的点都与发电站联通，求权重之和最小为多少。

思路：最小生成树，Prim算法初始选点时将发电站都选进去，Kruskal算法初始时把发电站都并到一起。

Prim算法代码：

/*************************************************************************     File Name: D.cpp     ID: obsolescence     BLOG: http://blog.csdn.net/obsolescence     LANG: C++      Mail: 384099319@qq.com      Created Time: 2016年07月27日 星期三 13时41分52秒 ************************************************************************/#include<bits/stdc++.h>#define Max(x,y) ((x)>(y)?(x):(y))#define Min(x,y) ((x)<(y)?(x):(y))#define each(it,v) for(__typeof((v).begin()) it=(v).begin();it!=(v).end();++it)#define Abs(x,y) ((x)>(y)?((x)-(y)):((y)-(x)))#define ll long long#define Mem0(x) memset(x,0,sizeof(x))#define Mem1(x) memset(x,-1,sizeof(x))#define MemX(x) memset(x,0x3f,sizeof(x))#define pb push_backusing namespace std;const int N=210,INF=0x3f3f3f3f;bool vis[N];int g[N][N],n,m,d[N];void Prim(){  Mem0(vis);  int i,j,sum=0;  for(i=1;i<=n;++i){    int a=-1,minn=INF;    for(j=1;j<=n;++j)      if(!vis[j] && d[j]<minn)        a=j,minn=d[j];    if(a==-1)break;    vis[a]=1;    sum+=minn;    for(j=1;j<=n;++j)      if(g[a][j] && !vis[j] && g[a][j]<d[j])d[j]=g[a][j];  }  printf("%d\n",sum);}int main(){  int t,nc,x,i,k,y,v;  while(~scanf("%d",&t)){    nc=0;    while(t--){      Mem0(g),MemX(d);      scanf("%d%d%d",&n,&m,&k);      while(k--){        scanf("%d",&x);        d[x]=0;      }      for(i=0;i<m;++i){        scanf("%d%d%d",&x,&y,&v);        g[x][y]=g[y][x]=v;      }      printf("Case #%d: ",++nc);      Prim();    }  }}

0 0