「WC2021」括号路径

最新推荐文章于 2024-09-12 19:02:20 发布

abigben

最新推荐文章于 2024-09-12 19:02:20 发布

阅读量332

点赞数

分类专栏：思维题文章标签：算法

本文链接：https://blog.csdn.net/qq_29361257/article/details/122095575

版权

思维题专栏收录该内容

17 篇文章 0 订阅

订阅专栏

题解

这道题有两个非常重要的性质需要发现 ~~其实挺显然的，不过好像大多人发现了没反应过来？~~：
$1 .$ 若 $(u, v)$ 是合法序列，那么 $(v, u)$ 也是合法序列。
$2 .$ 若 $(u, v), (v, k)$ 都是合法序列，那么 $(u, k)$ 也是合法序列 $\neq v \neq k)$ 。

根据性质 $2$ 可以发现：若 $i$ 和 $j$ 能构成合法路径，那么 $i$ 和 $j$ 能构成合法路径的点也一定能构成合法路径，相当于把 $i$ 合并到 $j$ 所在的集合里面。

综上，我们只需要每次将两个指向同一个集合的两个点所在的集合合并即可，合并时遵循启发式合并，效率即可优化到 $O (n l o g n)$ 。

题解

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+2e5+10;
inline int read(){
	int k=0,f=1;char c=getchar();
	while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
	while(c>='0'&&c<='9')k=k*10+c-'0',c=getchar();
	return k*f;
}
int n,m,k;
int fa[N],num[N];
map<int,int> e[N];
vector<int> col[N];
struct Node{int x,y,w,fst;};
inline int get(int x){return (x==fa[x])?x:fa[x]=get(fa[x]);}
queue<Node> q;
inline void add(int x,int y,int w){
	int kp=e[x][w];
	if(kp==y)return;
	if(!kp)e[x][w]=y,col[x].push_back(w);
	else q.push((Node){kp,y,w,x});
}
long long ans;
int main(){
	n=read();m=read();k=read();
	for(int i=1;i<=n;++i)fa[i]=i,num[i]=1;
	for(int i=1;i<=m;++i){
		int u=read(),v=read(),w=read();
		add(v,u,w);
	}
	while(q.size()){
		int x=q.front().x,y=q.front().y,w=q.front().w,fst=q.front().fst;q.pop();
		if(get(x)==get(y))continue;
		x=get(x);y=get(y);
		if(col[x].size()>col[y].size())swap(x,y);
		e[fst][w]=y;
		for(int j=0;j<(int)col[x].size();++j){
			int ver=col[x][j];
			add(y,e[x][ver],ver);
		}
		num[y]+=num[x];fa[x]=y;
	}
	for(int i=1;i<=n;++i)if(get(i)==i)ans+=1ll*num[i]*(num[i]-1)/2;
	printf("%lld\n",ans);
	return 0;
}