bzoj3712 [PA2014] Fiolki 并查集

题目链接：传送门

Description
化学家吉丽想要配置一种神奇的药水来拯救世界。
吉丽有n种不同的液体物质，和n个药瓶（均从1到n编号）。初始时，第i个瓶内装着g[i]克的第i种物质。吉丽需要执行一定的步骤来配置药水，第i个步骤是将第a[i]个瓶子内的所有液体倒入第b[i]个瓶子，此后第a[i]个瓶子不会再被用到。瓶子的容量可以视作是无限的。
吉丽知道某几对液体物质在一起时会发生反应产生沉淀，具体反应是1克c[i]物质和1克d[i]物质生成2克沉淀，一直进行直到某一反应物耗尽。生成的沉淀不会和任何物质反应。当有多于一对可以发生反应的物质在一起时，吉丽知道它们的反应顺序。每次倾倒完后，吉丽会等到反应结束后再执行下一步骤。
吉丽想知道配置过程中总共产生多少沉淀。

Input
第一行三个整数$n,m,k(0<=m<n<=200000,0<=k<=500000)$，分别表示药瓶的个数（即物质的种数），操作步数，可以发生的反应数量。
第二行有$n$个整数$g[1],g[2],\dots ,g[n]（1<=g[i]<=10^9)$，表示初始时每个瓶内物质的质量。
接下来$m$行，每行两个整数$a[i],b[i](1<=a[i],b[i]<=n,a[i]!=b[i])$，表示第$i$个步骤。保证$a[i]$在以后的步骤中不再出现。
接下来$k$行，每行是一对可以发生反应的物质$c[i],d[i](1<=c[i],d[i]<=n,c[i]!=d[i])$，按照反应的优先顺序给出。同一个反应不会重复出现。

Output
输出过程中产生沉淀的质量

注意到每种反应只会发生一次，因为一次反应至少会把其中一种物质反应完，之后就不会再有这种物质出现，也就不会再发生这种反应了qwq。
所以如果我们把所有珂能珂以进行的反应选出来，反应有先后顺序，因此求出这些反应是在什么时候进行并按此排序，得到的就是反应出现的顺序qwq
然后遍历所有珂以进行进行的反应，每次贪心地在两种物质中取min，更新物质的量并统计答案即可。
实现上，判断反应能否进行珂以用并查集，求出反应在什么时候进行就在并查集上每个点维护一个$tim$值，表示这个点向父亲连边的时间。
并查集不能路径压缩，用按秩合并，求$u,v$什么时候能反应就求$u$到$v$的链上的$tim$最大值（不包括$LCA$），这个值就表示什么时候$u,v$在同一个瓶子里qwq
然后注意如果两个反应的出现时间相同，要按题目给出的优先顺序排（被这个孙了一会……）

毒瘤代码

#include<stdio.h>
#include<cstring>
#include<algorithm>
#include<queue>
#define re register int
#define rl register ll
using namespace std;
typedef long long ll;
int read() {
	re x=0,f=1;
	char ch=getchar();
	while(ch<'0' || ch>'9') {
		if(ch=='-')	f=-1;
		ch=getchar();
	}
	while(ch>='0' && ch<='9') {
		x=(x<<1)+(x<<3)+ch-'0';
		ch=getchar();
	}
	return x*f;
}
inline void write(const int x) {
	if(x>9)	write(x/10);
	putchar(x%10+'0');
}
const int Size=200005;
int n,m,k,g[Size];
int father[Size],siz[Size],tim[Size],deep[Size];
int Find(int x) {
	if(x==father[x])	return x;
	int fa=Find(father[x]);
	deep[x]=deep[father[x]]+1;
	return fa;
}
void Union(int x,int y,int t) {
	x=Find(x);
	y=Find(y);
	if(x==y)	return;
	if(siz[x]<siz[y]) {
		father[x]=y;
		siz[y]+=siz[x];
		tim[x]=t;
	} else {
		father[y]=x;
		siz[x]+=siz[y];
		tim[y]=t;
	}
}
int Query(int u,int v) {
	if(Find(u)!=Find(v))	return -1;
	if(deep[u]<deep[v])	swap(u,v);
	int ans=0;
	while(deep[u]!=deep[v]) {
		if(tim[u]>ans)	ans=tim[u];
		u=father[u];
	}
	while(u!=v) {
		if(tim[u]>ans)	ans=tim[u];
		if(tim[v]>ans)	ans=tim[v];
		u=father[u];
		v=father[v];
	}
	return ans;
}
struct query {
	int c,d,id,t;
} w[500005];
inline bool comp(query x,query y) {
	if(x.t!=y.t)	return x.t<y.t;
	return x.id<y.id;
}
int main() {
	n=read();
	m=read();
	k=read();
	for(re i=1; i<=n; i++) {
		g[i]=read();
		father[i]=i;
		siz[i]=1;
	}
	for(re i=1; i<=m; i++) {
		int a=read();
		int b=read();
		Union(a,b,i);
	}
	int cnt=0;
	for(re i=1; i<=k; i++) {
		int c=read();
		int d=read();
		int t=Query(c,d);
		if(t!=-1) {
			w[++cnt]=(query){c,d,i,t};
		}
//		w[i].c=read();
//		w[i].d=read();
//		w[i].t=Query(w[i].c,w[i].d);
	}
	sort(w+1,w+1+cnt,comp);
	ll ans=0;
	for(re i=1; i<=cnt; i++) {
		int val=min(g[w[i].c],g[w[i].d]);
		g[w[i].c]-=val;
		g[w[i].d]-=val;
		ans+=val<<1;
	}
	printf("%lld",ans);
	return 0;
}
/*
9 5 8
1 2 4 3 3 3 4 3 1
8 1
6 9
2 5
5 9
7 9

2 6
4 7
5 7
1 8
4 3
2 8
4 9
4 5
*/