首页 > 代码库 > 最小生成树算法模板

最小生成树算法模板

#include<stdio.h>
#include<string.h>
#define Max 10000000
int map[110][110],cost[110],vit[110],m,n;
void prim(int x)
{
    int i,j,min,k,sum=0;
    memset(vit,0,sizeof(vit));
    for(i=1;i<=m;i++)
        cost[i]=map[x][i];
    cost[x]=0;
    vit[x]=1;
    for(i=1;i<m;i++)
    {
        min=Max;k=-1;
        for(j=1;j<=m;j++)
            if(!vit[j]&&cost[j]<min)
                min=cost[k=j];
            /*if(min==Max)
            {
                printf("?\n");
                return ;
            }*/
            vit[k]=1;
            sum+=min;
            for(j=1;j<=m;j++)
                if(!vit[j]&&cost[j]>map[k][j])
                    cost[j]=map[k][j];
    }
    printf("%d\n",sum);   
}