POJ - 2421 Constructing Roads【最小生成树】

题意：

给你n个城市，然后给出每个城市的距离。

再给你说已经有哪些城市已经连起来了。

剩下问你联通的最小修路代价是多少。

题解：

最小生成树。把连好的城市距离改成0就行了。

直接模板。

#include<iostream>
#include<queue>
#include<stdio.h>
#define ll long long

using namespace std;
const int maxn= 111;
int a[maxn][maxn];
int dis[maxn];
int n,m,x,y;
void prim(){
    int i,j,k,mina;
    bool f[maxn];
    for(i=2;i<=n;++i){
        f[i]=0;
        dis[i]=a[1][i];
    }
    dis[1]=0;
    f[1]=1;
    for(i=1;i<=n-1;++i){
        mina=9999999;
        k=0;
        for(j=1;j<=n;++j){
            if(!f[j]&&dis[j]<mina){
                mina=dis[j];
                k=j;
            }
        }
        if(k==0) return;
        f[k]=1;
        for(j=1;j<=n;++j){
            if(!f[j]&&a[k][j]!=9999999&&dis[j]>a[k][j]){
                dis[j]=a[k][j];
            }
        }
    }
}

int main(){
    while(~scanf("%d",&n)){
         for(int i=1;i<=n;++i)
            for(int j=1;j<=n;++j)
                scanf("%d",&a[i][j]);
         scanf("%d",&m);
         while(m--){
            scanf("%d %d",&x,&y);
            a[x][y]=a[y][x]=0;
         }
         prim();
         int ans=0;
         for(int i=1;i<=n;++i)
            ans+=dis[i];
        printf("%d\n",ans);
    }
    return 0;
}