hdu 1102 （最小生成树）

点击打开题目链接

该题就是一个简单的最小生成树问题，用的是prime算法。

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#define INF 0xffffff
#define MAX 110
int map[MAX][MAX],dis[MAX],visit[MAX];
int ans,n;
void prime(int node)
{
    memset(visit,0,sizeof(visit));
    for(int i=2;i<=n;i++) dis[i]=map[node][i];
    visit[node]=1; ans=0;
    for(int i=1;i<=n;i++)
    {
        int temp=INF,k;
        for(int j=1;j<=n;j++)
           if(!visit[j] && temp>dis[j])
              temp=dis[k=j];
        if(temp==INF) break;
        ans+=temp; visit[k]=1;
        for(int j=1;j<=n;j++)
            if(!visit[j] && dis[j]>map[k][j])
                dis[j]=map[k][j];
    }
}
int main()
{
    while(~scanf("%d",&n))
    {
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                scanf("%d",&map[i][j]);
        int m,a,b;
        ans=0;
        scanf("%d",&m);
        while(m--)
        {
            scanf("%d%d",&a,&b);
            map[a][b]=map[b][a]=0;
        }
        prime(1);
        printf("%d\n",ans);
    }
    return 0;
}