【POJ】[2421]Constructing Roads

这题的难度在于
题目已经帮助连通了一些村庄
要求其余的
不过只要直接把对应的村庄距离更新为0就好了

(-.-据说有人想到了并查集等等等等 然后超时了……)

#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 Q;
        scanf("%d",&Q);
        while(Q--) {
            int n,m;
            scanf("%d %d",&n,&m);
            e[n][m]=0;
            e[m][n]=0;
        }
        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
",sum);
    }
    return 0;
}

题目地址:【POJ】[2421]Constructing Roads