最短网络

题目大意：

Description

学校有n台计算机，为了方便数据传输，现要将它们用数据线连接起来。两台计算机被连接是指它们之间有数据线连接。由于计算机所处的位置不同，因此不同的两台计算机的连接费用往往是不同的。
当然，如果将任意两台计算机都用数据线连接，费用将是相当庞大的。为了节省费用，我们采用数据的间接传输手段，即一台计算机可以间接的通过若干台计算机（作为中转）来实现与另一台计算机的连接。
现在由你负责连接这些计算机，你的任务是使任意两台计算机都连通（不管是直接的或间接的）。

Input

输入文件wire.in，第一行为整数n（2<=n<=100），表示计算机的数目。此后的n行，每行n个整数。第x+1行y列的整数表示直接连接第x台计算机和第y台计算机的费用。

Output

输出文件wire.out，一个整数，表示最小的连接费用。

Sample Input

3
0 1 2
1 0 1
2 1 0

Sample Output

2（注：表示连接1和2，2和3，费用为2）

解题思路：

恩~挺简单的，就是普里姆算法~~这里提供两种（朴素算法O(n^3)和优化版O（n^2））

源程序注意了，Your（你们的） 重点：

先是优化版：

#include <cstdio>
#define r(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
int a[101][101],v[101],min,ans,n,p;
void read(int &f)//zb输入
{
    f=0;bool d=0;char c;
    while (c=getchar(),c<'0'||c>'9') if (c=='-') d=1;f=f*10+c-48;
    while (c=getchar(),c>='0'&&c<='9') f=f*10+c-48;
    if (d) f*=-1;
}
void write(int x)//zb输出
{
    if (x<0) {x=-x;putchar('-');}
    if (x) write(x/10);else return;
    putchar(x%10+48);
}
int main()
{
         read(n);
    r(i,1,n)
     r(j,1,n)
      read(a[i][j]);
    r(i,1,n)
     v[i]=a[1][i];
    r(k,1,n-1)
    {
        min=2147483647;
        r(i,1,n)
          if (v[i]<min&&v[i]!=0)
           min=v[i],p=i;
        ans+=v[p];      
        v[p]=0;
        r(i,1,n)
         if(a[p][i]<v[i])
          v[i]=a[p][i];
    }
    write(ans);
}

朴素算法：

#include <cstdio> //这种比较慢所以懒得浪费时间打zb输入输出
#define r(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
int a[101][101],v[101],min,ans,n,q,p;
int main()
{
    scanf("%d",&n);
    r(i,1,n)
     r(j,1,n)
      scanf("%d",&a[i][j]);
    v[1]=1;
    r(k,1,n-1)
    {
        min=2147483647;
        r(i,1,n)
         if(v[i]==1)
          r(j,1,n)
           if(a[i][j]<min&&a[i][j]!=0&&v[j]==0)
            min=a[i][j],p=j;
        if (min!=2147483647)
         ans+=min,v[p]=1;
    }
    printf("%d",ans);
}

转载于:https://www.cnblogs.com/Juruo-HJQ/p/9306969.html