算法训练营：等式

#include <iostream>
#include <vector>
using namespace std;

// ================= 代码实现开始 =================

const int N = 300005 ;

//Father : 每个节点的父亲节点
//Rank : 节点的秩
int Father[N] , Rank[N] ;

//查找节点x所有集合的根
// x : 节点x   返回值 ：根
int find( int x ){
    return Father[x] == x ? x : Father[x] = find( Father[x] ) ;
}

// 给定n个变量以及m个约束，判定是否能找出一种赋值方案满足这m个约束条件
// n：变量的个数
// m：约束条件的个数
// A：大小为m的数组，表示m条约束中的a
// B：大小为m的数组，表示m条约束中的b
// E：大小为m的数组，表示m条约束中的e
// 返回值：若能找出一种方案，返回"Yes"；否则返回"No"（不包括引号）。
string getAnswer(int n, int m, vector<int> A, vector<int> B, vector<int> E) {
    //初始化
    for( int i = 1 ; i <= n ; i++ )
    {
        Father[i] = i ;
        Rank[i] = 0 ;           //各个节点的秩起初都是0
    }
    // 将 e = 1 都放到 e = 0 前面
    int cnt = 0 ;
    for( int i = 0 ; i < m ; i++ )
    {
        if( E[i] == 1 )
        {
            swap( E[i] , E[cnt] ) ;
            swap( A[i] , A[cnt] ) ;
            swap( B[i] , B[cnt] ) ;
            cnt++ ;
        }
    }
    
    for( int i =0 ;  i< m ; i++ )
    {
        int setA = find( A[i] ) ;         //找到A[i]所有集合的根
        int setB = find( B[i] ) ;         //找到B[i]所有集合的根
        if( E[i] == 0 )
        {
            if( setA == setB )
                return "No" ;
        }
        else{
            if( setA != setB )
            {
                if( Rank[setA] > Rank[setB] )
                    swap( setA , setB ) ;
                //RankB 的父节点更高
                Father[setA] = setB ;
                if( Rank[setA] == Rank[setB] )
                    Rank[setB] ++ ;
            }
        }
    }
    return "Yes";
}


// ================= 代码实现结束 =================

int main() {
    int T;
    for (scanf("%d", &T); T--; ) {
        int n, m;
        scanf("%d%d", &n, &m);
        vector<int> A, B, E;
        for (int i = 0; i < m; ++i) {
            int a, b, e;
            scanf("%d%d%d", &a, &b, &e);
            A.push_back(a);
            B.push_back(b);
            E.push_back(e);
        }
        printf("%s\n", getAnswer(n, m, A, B, E).c_str());
    }
    return 0;
}