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;

}