poj2966--最小生成树Prim

poj2966--最小生成树Prim

最小生成树，Prim算法。
具体可参阅： http://www.cppblog.com/hoolee/archive/2012/08/06/186482.html
代码如下：

#include<stdio.h>
#include<stdlib.h>
#include<limits.h>
#define LEN 510
#define MAX 100000
int main()
{
    int i, j;
    int N, E;
    int A, B, K;
    int T;
    int mp[LEN][LEN];
    scanf("%d", &T);
    while(T--)
    {
        scanf("%d%d", &N, &E);
        for(i = 0; i < LEN; i++)
            for(j = i; j < LEN; j++)
                mp[i][j] = mp[j][i] = MAX;
        for(i = 0; i < E; i++)//read map
        {
            scanf("%d%d%d", &A, &B, &K);
            mp[B][A] = mp[A][B] = K;
        }
        int s[LEN] = {0};
        int cost[LEN];
        int lenall = 0;
        s[0] = 1;
        for(i = 0; i < N; i++)
            cost[i] = mp[0][i];
        for(j = 0; j < N - 1; j++)
        {
            int t = 0;
            int min = MAX;
            for(i = 0; i < N; i++)
                if(s[i] == 0 && cost[i] <= min)
                {
                    t = i;
                    min = cost[i];
                }
            s[t] = 1;
            lenall += min;
            for(i = 0; i < N; i++)
                if(s[i] == 0 && mp[t][i] < cost[i])
                    cost[i] = mp[t][i];
            
        }
        printf("%d\n", lenall);
    }
    //system("pause");
}