Graph Theory NO.5 HDU_1102_Constructing Roads_最小生成树

1. prim()

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define INF 0xffffff
int map[109][109];
int visit[1000009];
int low[100009];
int t,n,a,b;
int prim()
{
    for(int i=1;i<=n;i++)
    {
        low[i]=map[1][i];
        visit[i]=0;
    }
    visit[1]=1;
    int sum=0;
    for(int i=1;i<n;i++)
    {
        int Min=INF,pos;
        for(int j=1;j<=n;j++)
        {
            if(!visit[j]&&low[j]<Min)
            {
                Min=low[j];
                pos=j;
            }
        }
        visit[pos]=1;
        sum+=low[pos];
        for(int j=1;j<=n;j++)
        {
            if(!visit[j]&&low[j]>map[pos][j])
            {
                low[j]=map[pos][j];
            }
        }
    }
    return sum;
}
int main()
{
    while(~scanf("%d",&n))
    {
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                scanf("%d",&a);
                map[i][j]=a;
            }
        }
        scanf("%d",&t);
        for(int i=0;i<t;i++)
        {
            scanf("%d%d",&a,&b);
            map[a][b]=map[b][a]=0;
        }
        printf("%d\n",prim());
    }
}