Problem A

简单题意

给出村庄间的距离,并给出已经修好路的村庄,求使村庄连通起来最小修路距离

解题思路形成过程

这个题和Problem D畅通工程差不多,只不过这个已经有修好的一段路,看了网上一段代码的提示才突然发现,只要把已经修好路个两个村庄之间的距离设为0就行了,之后其他代码不变。

感想

就那么一点,怎么就没想到距离设为0呢

AC代码

#include <iostream>
#include <fstream>
#include <string.h>
using namespace std;
int n;
int Min[110];
int visit[110];
int dis[110][110];
void solve(){
    memset(visit,0,sizeof(visit));
    memset(Min,0x3f,sizeof(Min));
    int x=1;
    visit[1]=1;
    int sum=0;
    for(int i=2;i<=n;i++)
        Min[i]=min(Min[i],dis[1][i]);
    for(int i=1;i<n;i++){
        int temp=10000000;
        for(int j=1;j<=n;j++)
        if(!visit[j]&&temp>Min[j]){
            temp=Min[j];
            x=j;
        }
        sum+=temp;
        visit[x]=1;
        for(int j=1;j<=n;j++)
            if(!visit[j]&&dis[j][x]<Min[j])
              Min[j]=dis[j][x];
    }
    cout<<sum<<endl;
}
int main()
{
    ifstream cin("in.txt");
    int m,a,b;
    while(cin>>n){
        memset(dis,0,sizeof(dis));
        for(int i=1;i<=n;i++)
           for(int j=1;j<=n;j++)
               cin>>dis[i][j];
               cin>>m;
        for(int i=0;i<m;i++){
            cin>>a>>b;
            dis[a][b]=0;dis[b][a]=0;            
        }
        solve();
    }
    return 0;
}

你可能感兴趣的:(Problem A)