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;
}