poj 1258 prim

显然是prim

#include<stdio.h>
#include<cstring>
const int inf=999999;
int map[101][101],dis[101],book[101];
int main()
{
    int n;
    while(scanf("%d",&n)!=EOF)                                //这里忘记！=EOF的话TLE
    {
        memset(map,inf,sizeof(map));
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<n;j++)
            {
                scanf("%d",&map[i][j]);
            }
        }
        for(int i=0;i<n;i++)
        {
            dis[i]=map[0][i];
            book[i]=0;
        }
        book[0]=1;
        int sum=0;
        for(int q=1;q<n;q++)
        {
            int k=-1;
            int min=inf;
            for(int i=0;i<n;i++)
            {
                if(book[i]==0&&dis[i]<min)
                {
                    min=dis[i];
                    k=i;
                }
            }
            book[k]=1;
            sum+=min;
            for(int i=0;i<n;i++)
            {
                if(book[i]==0&&map[k][i]<dis[i])
                    dis[i]=map[k][i];
            }
        }
        printf("%d\n",sum);
    }
    return 0;
}