【杭电】[1879]继续畅通工程

如果道路已建
则把成本赋值为0

#include<stdio.h>
#include<algorithm>
using namespace std;
int dis[120];
int e[120][120];
bool flag[120];
int inf=99999999;
int main() {
    int n;
    while(scanf("%d",&n),n) {
        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;
            }
        }
        int m=n*(n-1)/2;
        while(m--) {
            int i,j,t,tflag;
            scanf("%d %d %d %d",&i,&j,&t,&tflag);
            if(tflag)
                e[i][j]=e[j][i]=0;
            else if(e[i][j]>t) {
                e[i][j]=t;
                e[j][i]=t;
            }
        }
        int sum=0;
        dis[1]=0;
        bool x=true;
        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;
            else {
                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;
}

题目地址:【杭电】[1879]继续畅通工程