【杭电】[1102]Constructing Roads

最小生成树模板题

先以矩阵形式给了村庄与村庄之间道路的距离
然后再给出哪些村庄之间道路已经通了
此时把这两个村庄的距离改为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)!=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++) {
                int t;
                scanf("%d",&t);
                e[i][j]=t;
            }
        }
        int m;
        scanf("%d",&m);
        while(m--) {
            int ts,te;
            scanf("%d %d",&ts,&te);
            e[ts][te]=e[te][ts]=0;
        }
        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;
}

题目地址:【杭电】[1102]Constructing Roads