2016sdau课程练习专题四 1001

1.题目编号

1001

2.简单题意

有n个村庄，编号1-n，以矩阵的形式给出任意两个村庄之间的距离，然后告诉已经有q个村庄已经修好了路，问现在要打算使所有村庄都联通需要修路的最小长度

3.解题思路

最小生成树的小变种，如果将已经存在的路距离修改成0，这样就会保证用prim生成树时这条路一定会入选并且<span style="font-family: Arial, Helvetica, sans-serif;">长度不会计算在最小生成树内。</span>

4.感想

最小树的求解问题，比较难做，没有做过啊，但还是好好做吧！ 

5.代码

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=110;
const int INF=0x3f3f3f3f;
int n,ans;
int map[N][N],dis[N],vis[N];
void Prim()
{
    int i;
    for(i=1;i<=n;i++)
{
        dis[i]=map[1][i];
        vis[i]=0;
    }
    dis[1]=0;
    vis[1]=1;
    int j,k,tmp;
    for(i=1;i<=n;i++){
        tmp=INF;
        for(j=1;j<=n;j++)
            if(!vis[j] && tmp>dis[j]){
                k=j;
                tmp=dis[j];
            }
        if(tmp==INF)
            break;
        vis[k]=1;
        ans+=dis[k];
        for(j=1;j<=n;j++)
            if(!vis[j] && dis[j]>map[k][j])
                dis[j]=map[k][j];
    }
}


int main()
{
    while(~scanf("%d",&n)){
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                scanf("%d",&map[i][j]);
        int q,a,b;
        scanf("%d",&q);
        while(q--){
            scanf("%d%d",&a,&b);
            map[a][b]=map[b][a]=0;
        }
        ans=0;
        Prim();
        printf("%d\n",ans);
    }
    return 0;
}