leetcode六月每日一题 leetcode 990

这一题利用并查集（可以高效的回答连接类的问题）来解决

首先遍历所有的等式，构造并查集。同一个等式中的两个变量属于同一个连通分量，因此将两个变量进行合并。

然后遍历所有的不等式。同一个不等式中的两个变量不能属于同一个连通分量，因此对两个变量分别查找其所在的连通分量，如果两个变量在同一个连通分量中，则产生矛盾，返回 false。

如果遍历完所有的不等式没有发现矛盾，则返回 true。

class UnionFind{

private:
    vector<int> save;

public:
    UnionFind(){
        save.resize(26);
        iota(save.begin() , save.end(), 0); //自增的赋值
    }

    int find( int index ){
        if( index == save[index] ) return index; //等于本身的话，证明为根节点
        else{ //沿着当前变量父节点一直向上查找
            save[index] = find( save[index] );
            return save[index];
        }
    }

    void union_elements( int index1, int index2 ){
        save[ find(index1) ] = find(index2);
    }

};

class Solution {
public:
    bool equationsPossible(vector<string>& equations) {

        UnionFind uf;
        for(const string& str : equations){
            if(str[1] == '='){
                int index1 = str[0] - 'a';
                int index2 = str[3] - 'a';
                uf.union_elements(index1,index2);
            }
        }

        for(const string& str : equations){
            if(str[1] == '!'){
                int index1 = str[0] - 'a';
                int index2 = str[3] - 'a';
                if(uf.find(index1) == uf.find(index2))
                    return false;
            }
        }

        return true;
    }
};