Luogu P1396 营救

Luogu P1396 营救

本来刚开始想用Dijkstra，但调了一半天都还有问题，所以果断放弃。
后来考虑了一下题目，发现可以考虑求最小生成树，于是开Kruskal。
果然最小生成树还是很好想的。

#include<bits/stdc++.h>
#define N 10010
#define M 20010

using namespace std;

int n,m,s,t,cnt,tot,ans;
int fa[N];

struct node {
	int frm,to,val;
}edge[M];

void addEdge(int u,int v,int w) {
	edge[++cnt]=(node){u,v,w};
	return;
}

bool cmp(node a,node b) {
	return a.val<b.val;
}

int Find(int x) {
	return fa[x]==x?x:fa[x]=Find(fa[x]);
}

bool Merge(int x,int y) {
	return Find(x)==Find(y)?false:true;
}

void Union(int x,int y) {
	fa[Find(y)]=Find(x);
	return;
}

void Read() {
	scanf("%d%d%d%d",&n,&m,&s,&t);
	for(int i=1;i<=m;i++) {
		int u,v,w;
		scanf("%d%d%d",&u,&v,&w);
		addEdge(u,v,w);
	}
	return;
}

void Init() {
	for(int i=1;i<=n;i++) {
		fa[i]=i;
	}
	return;
}

void Kruskal() {
	sort(edge+1,edge+cnt+1,cmp);
	for(int i=1;i<=cnt;i++) {
		if(Merge(edge[i].frm,edge[i].to)) {
			Union(edge[i].frm,edge[i].to);
			if(Merge(s,t)) {
				ans=max(ans,edge[i].val);
			}
			else {
				ans=max(ans,edge[i].val);
				break;
			}
		}
	}
	return;
}

void Print() {
	printf("%d",ans);
	return;
}

int main()
{
	Read();
	Init();
	Kruskal();
	Print();
	return 0;
}