POJ 1258 Agri-Net(裸prim)

有多个点，让你找出将所有点联通的最短路径，直接裸prim，注意是多组数据。

#include<stdio.h>
#define maxn 200000
int N,i,j,k,sum;
int weight[110][110],distance[110];//distance数组表示点到生成树的距离
int record[110];//记录该点是否被访问
int findmin();//每次找离生成树最近的一个点
int renew(int x);//加入该点后进行更新
int main()
{
    while(scanf("%d",&N)!=EOF)
    {
        sum=0;
        for(i=1;i<=N;i++)
            record[i]=0;
        for(i=1;i<=N;i++)
            for(j=1;j<=N;j++)
                scanf("%d",&weight[i][j]);
        //随意找一个点开始(第一个点),并进行预处理
        for(i=1;i<=N;i++)
            distance[i]=weight[1][i];
        record[1]=1;

        for(i=2;i<=N;i++)//还要加入N-1个点
        {
            j=findmin();//j是当前到生成树最近的点，加入生成树
            record[j]=1;
            sum+=distance[j];
            renew(j);
        }
        printf("%d\n",sum);
    }
    return 0;
}

int findmin()
{
    int i,j,k;
    j=200000;
    for(i=2;i<=N;i++)
        if(!record[i]&&distance[i]<j)//该点未被访问且最小
        {
            k=i;
            j=distance[i];
        }
    return k;
}

int renew(int x)
{
    for(int i=1;i<=N;i++)
        if(!record[i]&&distance[i]>weight[x][i])//如果通过点x比直接到生成树要近就进行更新
            distance[i]=weight[x][i];
}