【POJ】[1258]Agri-Net

现在看都是非常水的一个最小生成树了……
可以说是秒A……

#include<stdio.h>
#include<algorithm>
using namespace std;
int main() {
    int dis[120];
    int e[120][120];
    bool flag[120];
    int N,inf=999999;
    while(scanf("%d",&N)!=EOF) {
        for(int i=1; i<=N; i++) {
            dis[i]=inf;
            flag[i]=false;
        }
        for(int i=1; i<=N; i++) {
            for(int j=1; j<=N; j++) {
                e[i][j]=inf;
            }
        }
        for(int i=1; i<=N; i++) {
            for(int j=1; j<=N; j++) {
                if(i==j) {
                    int t;
                    scanf("%d",&t);
                } else
                    scanf("%d",&e[i][j]);
            }
        }
        int sum=0;
        dis[1]=0;
        while(true) {
            int v=-1;
            for(int u=1; u<=N; u++) {
                if(!flag[u]&&(v==-1||dis[u]<dis[v]))
                    v=u;
            }
            if(v==-1)
                break;
            flag[v]=true;
            sum+=dis[v];
            for(int u=1; u<=N; u++) {
                dis[u]=min(dis[u],e[v][u]);
            }
        }
        printf("%d\n",sum);
    }
    return 0;
}

题目地址:【POJ】[1258]Agri-Net