zoj1372

最新推荐文章于 2018-09-26 15:19:00 发布

原创最新推荐文章于 2018-09-26 15:19:00 发布 · 217 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#zoj

zoj 专栏收录该内容

201 篇文章

订阅专栏

本文介绍使用Prim算法解决给定点与路线情况下求最小生成树的问题，通过具体实现代码展示了如何找到连接所有节点的最短路径。

题目大意：

给P个点，R条路线，每条路线给连接的两端和cable长度。
输出连接全部端点的总长度

解题思路：

prim算法应用

代码如下：

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<stdlib.h>

using namespace std;

int p,r;
int map[60][60];
bool node[60];

int prim()
{
  int lenn=0,num=1,i,j,max;
  int nd1,nd2;
  node[1]=true;
  while(num<p)
  {
    max=200;
    for(i=1;i<=p;i++)
    {
      if(node[i])
      {
        for(j=1;j<=p;j++)
        {
          if(!node[j]&&(map[i][j]<101)&&(map[i][j]<max))
          {
            max=map[i][j];nd1=i;nd2=j;
          }
        }
      }
    }
    num++;
    node[nd2]=true;
    lenn+=map[nd1][nd2];
  }  
  return lenn;
}

int main()
{
  scanf("%d%d",&p,&r);
  while(p)
  {
    if(p==0)
      break;
    int i,j,a,b,c;
    memset(map,101,sizeof(map));
    for(i=1;i<=p;i++)
      node[i]=false;
    for(i=1;i<=r;i++)
    {
      scanf("%d%d%d",&a,&b,&c);
      if(c<map[a][b])
      {
        map[a][b]=map[b][a]=c;
      }
    }
    printf("%d\n",prim());
    scanf("%d%d",&p,&r);
  }
  return 0;
}