ZOJ 3204 Connect them 最小生成树+字典序最小

点击打开链接

有N个点，每两个点要想连通都有一个花费，有些点之间不能连通，让你建造一些路，使得所有点能够连通。总的路程必须最短，如果有多个最短，选出使得输出字典序最小的路。

最小生成树的变式，还是用kruskal，排序的时候按照字典序排一下，输出的时候也按照字典序排一下就好了。注意双向的路都要加进去，比如2-3和3-2两条路都要有，这样才能保证字典序。

 

#include<bits/stdc++.h>
using namespace std;
struct Edge{
	int s,e,w;
	Edge(int _s,int _e,int _w):s(_s),e(_e),w(_w){}
	bool operator <(const Edge &v)const{
		//如果权值相同，字典序小的优先 
		if(w!=v.w) return w<v.w; 
		if(s!=v.s) return s<v.s;
		return e<v.e;
	}
};
typedef pair<int,int> P; 
vector<Edge> edges;
vector<int> par;
vector<P> res; //排序 
int root(int a)
{
	return par[a]==a?a:par[a]=root(par[a]);
}
void merge(int a,int b)
{
	int p1=root(a);
	int p2=root(b);
	if(p1!=p2) par[p2]=p1;
}
int main()
{
//	freopen("E:\\ACM\\test.txt","r",stdin);
	int T;scanf("%d",&T);
	while(T--)
	{
		edges.clear();res.clear();par.clear();
		int n;scanf("%d",&n);
		for(int i=0;i<=n;i++) par.push_back(i);
		int w;
		for(int i=1;i<=n;i++)
			for(int j=1;j<=n;j++)	
			{
				scanf("%d",&w);
				if(w) edges.push_back(Edge(i,j,w)); 
			}
		sort(edges.begin(),edges.end());
		int cnt=0;
		for(int i=0;i<edges.size();i++)
		{
			if(root(edges[i].s)!=root(edges[i].e))
			{
				merge(edges[i].s,edges[i].e);
				res.push_back(P(edges[i].s,edges[i].e));
				++cnt;
			}
			if(cnt==n-1) break;
		}
		if(cnt!=n-1) puts("-1");
		else
		{
			sort(res.begin(),res.end());
			for(int i=0;i<res.size();i++)
				printf("%d %d%c",res[i].first,res[i].second,i==res.size()-1?'\n':' ');
		}
	}	
	
	return 0;
}