NYOJ筹建工程之并查集and最小生成树prim算法

最新推荐文章于 2018-10-11 17:04:46 发布

原创最新推荐文章于 2018-10-11 17:04:46 发布 · 373 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种结合并查集和Prim算法解决特定图论问题的方法。通过并查集处理连通性问题，并利用Prim算法求解最小生成树，适用于解决包含多个连通分量的加权无向图问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

http://acm.nyist.net/JudgeOnline/problem.php?pid=502

//并查集and最小生成树prim算法

#include<stdio.h>
#include<string.h>
#define MAX 0x3f3f3f
#define N 110
int graph[N][N];
int visited[N],ps[N];
int a[N];
int find(int k) //并查集
{
int r=k;
while(a[r]!=r)
r=a[r];
int y=k,tem;
while(y!=r) //路径压缩
{
  tem=a[y];
  a[y]=r;
  y=tem;
}
return r;
}
int cmp(int m)   //prim 算法
{
int i,j,pos,min,sum=0;
memset(visited,0,sizeof(visited));
pos=1;
visited[1]=1;
for(i=2;i<=m;i++)
ps[i]=graph[pos][i];

for(i=1;i<m;i++)
{
  min=MAX;
  for(j=1;j<=m;j++)
  {
   if(!visited[j]&&ps[j]<min)
   {
    min=ps[j];
    pos=j;
   }
  }
  sum+=min;
  visited[pos]=1;   //进行标记
  for(j=1;j<=m;j++)
  if(!visited[j]&&graph[pos][j]<ps[j])//更新权值
  ps[j]=graph[pos][j];
}
printf("%d\n",sum);
return 0;
}
int main()
{
int T,n,m,i,j;
int x,y,money;
int t1,t2,s;
scanf("%d",&T);
while(T--)
{
  s=0;
  memset(graph,MAX,sizeof(graph));
  scanf("%d%d",&n,&m);
  for(i=1;i<=m;i++)
  a[i]=i;
  for(i=1;i<=n;i++)
  {
   scanf("%d%d%d",&x,&y,&money);
   graph[x][y]=graph[y][x]=money;
   t1=find(x);
   t2=find(y);
   if(t1!=t2)
   a[t1]=t2;
  }

  for(i=1;i<=m;i++)
  if(a[i]==i)
  s++;
  if(s>=2)
  {
  printf("No solution\n");
  continue;
     }
     cmp(m);
}
return 0;
}