[loj3462]括号路径

对于两条边$(x_{1},y,c)$和$(x_{2},y,c)$，不难发现$x_{1}$与$x_{2}$完全等价，因此可以合并

重复此过程，合并之后用启发式合并来合并边集（注意自环也可以参与合并，即$(x,y,c)$和$(y,y,c)$，那么就将$x$和$y$合并），并用一个队列来记录操作

之后，当不存在此类边，考虑一条合法路径中必然有相邻两次，选择了同类型的左右括号（先左后右），由于不存在此类边，不难证明这只能在一条边上往返，显然无意义，即只能在合并后的集合内部选，并查集维护即可

#include<bits/stdc++.h>
using namespace std;
#define N 300005
struct ji{
    int x,c,y;
};
queue<ji>q;
map<int,int>mat[N];
map<int,int>::iterator it;
int n,m,k,x,y,z,f[N],sz[N];
long long ans;
int find(int k){
    if (k==f[k])return k;
    return f[k]=find(f[k]);
}
void merge(int x,int y){
    x=find(x),y=find(y);
    if (x==y)return;
    if (mat[x].size()<mat[y].size())swap(x,y);
    f[y]=x;
    sz[x]+=sz[y];
    for(it=mat[y].begin();it!=mat[y].end();it++)
        q.push(ji{x,(*it).first,(*it).second});
}
int main(){
    scanf("%d%d%d",&n,&m,&k);
    for(int i=1;i<=n;i++){
        f[i]=i;
        sz[i]=1;
    } 
    for(int i=1;i<=m;i++){
        scanf("%d%d%d",&x,&y,&z);
        q.push(ji{y,z,x});
    }
    while (!q.empty()){
        ji o=q.front();
        q.pop();
        o.x=find(o.x);
        if (!mat[o.x][o.c])mat[o.x][o.c]=o.y;
        else merge(mat[o.x][o.c],o.y);
    }
    for(int i=1;i<=n;i++)
        if (f[i]==i)ans+=1LL*sz[i]*(sz[i]-1)/2;
    printf("%lld",ans);
}