wust oj-1943最小生成树 邻接矩阵Prim

1943: 最优布线问题

Time Limit: 1 Sec   Memory Limit: 128 MB   64bit IO Format: %lld
Submitted: 43   Accepted: 21
[ Submit][ Status][ Web Board]

Description

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

Input

多组测试数据。
第一行为整数n(2<=n<=100),表示计算机的数目。
此后的n行,每行n个整数。第x+1行y列的整数表示直接连接第x台计算机和第y台计算机的费用。

Output

输出一个整数,表示最小的连接费用。

Sample Input 

3
0 1 2
1 0 1
2 1 0

Sample Output

2

一道Prim的入门模板题

#include<iostream>
#include<cstring>
using namespace std;
const int INF=0x3f3f3f3f;
int a[105][105];
int vis[105];
int mincost[105];
int t;
int Prim()
{
    int k;
    int sum=0;
    for(int i=0;i<t;i++)
        mincost[i]=a[0][i];   //初始化mincost数组,求出从0点出发的最小边
    vis[0]=1;//把0点标为已选
    for(int i=0;i<t;i++)//从1开始找,0已找过
    {
        int mmin=INF;       //把mmin置为无穷大,找出下一个最短的边
        for(int j=0;j<t;j++)
            if(mmin>mincost[j]&&!vis[j])
        {
            mmin=mincost[j];        //更新
            k=j;            //k记录最短边的编号
        }
        vis[k]=1;           //把点k标记为已选
        if(mmin<INF) sum+=mmin;
        //更新mmincost数组,现在表示从k出发的最短边
        for(int j=0;j<t;j++)
            if(!vis[j]&&mincost[j]>a[k][j])     //有些mincost未被更新,是原来节点指出的最短边,更新后的是从现在结点指出的
        {
            mincost[j]=a[k][j];
        }
    }
    return sum;
}
int main()
{
    while(cin>>t)
    {
        memset(vis,0,sizeof vis);
        memset(mincost,0,sizeof mincost);
        for(int i=0;i<t;i++)
            for(int j=0;j<t;j++)
            cin>>a[i][j];
        cout<<Prim()<<endl;

    }

    return 0;
}

猜你喜欢

转载自blog.csdn.net/codetypeman/article/details/80686868