POJ_1258_Agri-Net(最小生成树 Prim)

还是注意初始化和数组的大小。跟前几题没什么区别。

# include <stdio.h>
# include <string.h>
# define inf 100010
int map[120][120];
int m;
int Prim()
{
    int low[120];
    int visit[120]={0};
    int i,j,k=1;
    int s=1;
    int point;
    int min;
    int sum=0;
    for(i=1;i<=m;i++)
        low[i]=inf;
    visit[s]=1;
    while(1)
    {
        if(k == m)
            break;
        min=inf;
        for(j=2;j<=m;j++)
        {
            if(!visit[j] && low[j] > map[s][j])
                low[j]=map[s][j];
            if(!visit[j] && min>low[j])
            {
                min=low[j];
                point=j;
            }
        }
        sum+=min;
        s=point;
        visit[s]=1;
        k++;
    }
    return sum;
}
int main()
{
    int i,j,k;
    while(scanf("%d",&m) != EOF)
    {
    
        for(i=1;i<=m;i++)
        {
            for(j=1;j<=m;j++)
            {
                scanf("%d",&map[i][j]);
            }
        }
        printf("%d\n",Prim());
    }
    return 0;
}