等式方程的可满足性

LeetCode算法网站算法题

https://leetcode-cn.com/problems/satisfiability-of-equality-equations/

思路：

并查集的应用，把所有相等的节点并入同一个集合，即拥有相同祖先，然后在遍历不相等的变量，如果两个不相等的变量仍然存在于相等的那一个集合之中，则返回错误。

class union_set
{
private:
    vector<int>parents;
public:
    union_set();//初始化并查集数据
    int find_parents(int index);//寻找节点的最终父节点
    void unite(int index1,int index2);//整合两个节点
};
union_set::union_set()
{
    int i;
    for(i=0; i<26; i++)
        parents.push_back(i);//初始化所有节点，把所有节点的祖先设置为自己
}
int union_set::find_parents(int index)
{
    if(index==parents[index])
        return index;
    parents[index]=find_parents(parents[index]);//所以在这里就能明白为什么要改变当前节点的祖先值了，因为如果当前节点的祖先节点都有了别的祖先节点，那么就需要把当前节点的祖先节点改变为原来祖先节点的祖先节点
                                                //还有很重要的一点是，有可能当前节点在并查集合并的时候没有更改祖先节点，但是在后来判断的时候也可以更改自己的祖先节点，这样判断的结果仍然是正确的
    return parents[index];
}
void union_set::unite(int index1,int index2)
{
    parents[find_parents(index1)]=find_parents(index2);//把index1的祖先节点的祖先由原来祖先的祖先赋值为index2的祖先节点
}
class Solution
{
public:
    bool equationsPossible(vector<string>& equations)
    {
        union_set us;
        for (const string& str: equations)
        {
            if (str[1] == '=')
            {
                int index1 = str[0] - 'a';
                int index2 = str[3] - 'a';
                us.unite(index1, index2);//连接两个相等的节点
            }
        }
        for (const string& str: equations)
        {
            if (str[1] == '!')
            {
                int index1 = str[0] - 'a';
                int index2 = str[3] - 'a';
                if (us.find_parents(index1) == us.find_parents(index2))//如果不应该相等的两个元素，在同一个相等集合中（即拥有公共祖先），就返回假
                {
                    return false;
                }
            }
        }
        return true;
    }
};