洛谷 1525 关押罪犯

一道并查集，按照一般的思路是不能处理某一些情况的。

例如

1——2 3——4 1——5 3——5 在安排第二个关系的时候不确定把3放到1还是放到2，而这个关系的放置会直接影响第三个关系。当时想了好久都没有思路， 后来看了别人的代码，是用一个数组记录敌人的信息。这个用的真的很巧妙，用了这个数组之后，对于第二个关系来说，他们和第一个都没有关系，那着两个关系就不要合并，就避免了合并产生的错误。

//关押罪犯
#include <bits/stdc++.h>
using namespace std;
const int maxn=100000+5;
int fa[20005];
int N,M;
int enemy[20005];
struct Node{
	int a,b,c;
	bool operator<(const Node x)const{
		return this->c<x.c;
	}
}; 

int find(int x)
{
	if(x==fa[x]) return x;
	else return fa[x]=find(fa[x]);
}

void unit(int x,int y)
{
	x=find(x);
	y=find(y);
	if(x==y) return;
	fa[x]=y;
}

void init()
{
	for(int i=1;i<=N;i++)
		fa[i]=i;
}

int main()
{
	priority_queue<Node> q;
	int a,b,c;
	cin>>N>>M;
	int i,j,k;
	for(i=0;i<M;i++){
		scanf("%d%d%d",&a,&b,&c);
		q.push((Node){a,b,c});
	}
	Node temp;
	init();
	while(!q.empty())
	{
		temp=q.top();
		q.pop();
		if(find(temp.a)==find(temp.b)){
			printf("%d",temp.c); 
			return 0;
		}	
		if(!enemy[temp.a]) enemy[temp.a]=temp.b;
		else unit(enemy[temp.a],temp.b);
		if(!enemy[temp.b]) enemy[temp.b]=temp.a;
		else unit(enemy[temp.b],temp.a);
	} 
	printf("0");
	return 0;
}