hdoj 1102 Constructing Roads

题目：http://acm.hdu.edu.cn/showproblem.php?pid=1102

分析：看到这题给出的都是矩阵形式，就知道了可以用Prim算法求MST。

#include <iostream>



using namespace std;

#define typec int

#define V 101

const typec inf=0x3f3f3f3f;

int vis[V];

typec lowc[V],cost[V][V];

/*=================================================*\

 | Prim求MST

 | INIT: cost[][]耗费矩阵(inf为无穷大);

 | CALL: prim(cost, n); 返回-1代表原图不连通;

\*==================================================*/

typec prim(int n)

{

    int i,j,p;

    typec minc,res=0;

    memset(vis,0,sizeof(vis));

    vis[0]=1;

    for(i=1; i<n; i++) lowc[i]=cost[0][i];

    for(i=1; i<n; i++)

    {

        minc=inf;

        p=-1;

        for(j=0; j<n; j++)

        {

            if(0==vis[j]&&minc>lowc[j])

            {

                minc=lowc[j];

                p=j;

            }

        }

        if(inf==minc) return -1;

        res+=minc;

        vis[p]=1;

        for(j=0; j<n; j++)

            if(0==vis[j]&&lowc[j]>cost[p][j])lowc[j]=cost[p][j];

    }

    return res;

}





int main()

{

 //   freopen("input.txt","r",stdin);

    int n,m,i,j,u,v;

    while(cin>>n&&n!=0)

    {

        memset(cost,inf,sizeof(cost));

        for(i=0; i<n; i++)

            for(j=0; j<n; j++)

                cin>>cost[i][j];

        cin>>m;

        for(i=0; i<m; i++)

        {

            cin>>u>>v;

            cost[u-1][v-1]=cost[v-1][u-1]=0;

        }

        cout << prim(n) << endl;

    }

    return 0;

}