HDU 1102 Constructing Roads 基础最小生成树★

题意 :给你N个村庄,接下来是N行输入,第 i 行的第 j个数字代表的是编号为 i的村庄到编号为j的村庄的距离,接下来又有Q行输入,每行两个整数x, y.表示x和y之间已经修好路了.问要将所有村庄连在一起,最少要修多长的公路。

 

分析：把已经修好的路连接，然后再未修的路中选一些生成最小生成树。

代码：

#include<iostream>#include<algorithm>#include<queue>#include<vector>#include<cstring>#include<cstdio>#include<cmath>const int INF=0x3f3f3f3f;typedef long long LL;using namespace std;const int maxn=1005;struct Edge{int s,e; //起点，终点，权值 int w;Edge(){}Edge(int _s,int _e,int _w):s(_s),e(_e),w(_w){}bool operator <(const Edge &v)const{return w<v.w; } };vector<Edge> edges;vector<int> par;int N;int GetRoot(int a){return par[a]==a?a:par[a]=GetRoot(par[a]);}void Merge(int a,int b){int p1=GetRoot(a);int p2=GetRoot(b);if(p1==p2) return;par[p2]=p1;}int main(){//freopen("E:\\ACM\\test.txt","r",stdin);while(cin>>N){edges.clear();par.clear();for(int i=0;i<=N;i++) par.push_back(i);int w,Q,a,b;for(int i=1;i<=N;i++) {for(int j=1;j<=N;j++){cin>>w;edges.push_back(Edge(i,j,w));}}cin>>Q; //已经修好了Q条道路 for(int i=0;i<Q;i++) {cin>>a>>b;Merge(a,b);}sort(edges.begin(),edges.end()); //边权值按小到大排序 int cnt=Q; //表示已经连好了Q条边 int ans=0;for(int i=0;i<edges.size();i++){if(GetRoot(edges[i].s)!=GetRoot(edges[i].e)){Merge(edges[i].s,edges[i].e);ans+=edges[i].w;}if(cnt==N-1) break;}cout<<ans<<endl;}return 0;}