优化之后的并査集查找之“ 等式方程的可满足性”

题目：力扣990. 等式方程的可满足性

给定一个由表示变量之间关系的字符串方程组成的数组，每个字符串方程 equations[i] 的长度为 4，并采用两种不同的形式之一：“a==b” 或 “a!=b”。在这里，a 和 b 是小写字母（不一定不同），表示单字母变量名。
只有当可以将整数分配给变量名，以便满足所有给定的方程时才返回 true，否则返回 false。

示例 1：
输入：["a==b","b!=a"]
输出：false
解释：如果我们指定，a = 1 且 b = 1，那么可以满足第一个方程，但无法满足第二个方程。没有办法分配变量同时满足这两个方程。
示

来源：力扣（LeetCode）

解题思路

未优化之前执行时间超过88%，优化之后超越100%
通过并査集判断两个字母是否相等。先把所有相等的判断完，再去判断不相等的。相等的话直接放到同一个并査集里面，不相等的话，判断是否位于一个并査集。见代码注释。

代码

class Solution(object):
    def equationsPossible(self, equations):
        """
        通过并査集判断两个字母是否相等
        先把所有相等的判断完，再去判断不相等的
        相等的话直接放到同一个并査集里面，不相等的话，判断是否位于一个并査集
        :type equations: List[str]
        :rtype: bool
        """
        parents = [chr(i+97) for i in range(26)]  # ord('a') = 97 parents[0]表示'a'的父节点，初始化指向自己'a'
        sizes = [1 for _ in range(26)]  # 每棵树的节点总数

        # def find_parent(char1):
        #     while char1 != parents[ord(char1)-97]:
        #         char1 = parents[ord(char1)-97]
        #     return char1

        # 优化：采用路径压缩的方法：降低并査集所形成的的树的高度
        def find_parent(char1):
            while char1 != parents[ord(char1)-97]:
                parents[ord(char1) - 97] = parents[ord(parents[ord(char1)-97])-97]  # parent(x) = parent(parent(x))
                char1 = parents[ord(char1) - 97]
            return char1

        def is_same_parent(char1, char2):
            char1 = find_parent(char1)
            char2 = find_parent(char2)
            return True if char1 == char2 else False

        # def union(char1, char2):
        #     char1 = find_parent(char1)
        #     char2 = find_parent(char2)
        #     if char1 != char2:
        #         parents[ord(char1)-97] = char2
        #     return

        # 优化：小的树向大的树合并
        def union(char1, char2):
            char1 = find_parent(char1)
            char2 = find_parent(char2)
            if char1 != char2:
                if sizes[ord(char1)-97] > sizes[ord(char2)-97]:
                    parents[ord(char2)-97] = char1
                    sizes[ord(char1)-97] += sizes[ord(char2)-97]
                else:
                    parents[ord(char1) - 97] = char2
                    sizes[ord(char2) - 97] += sizes[ord(char1) - 97]
            return

        for s in equations:
            if s[1] == '=':
                char1 = s[0]
                char2 = s[3]
                union(char1, char2)

        for s in equations:
            if s[1] == '!':
                char1 = s[0]
                char2 = s[3]
                if is_same_parent(char1,char2):
                    return False
        return True

# s = Solution()
# print(s.equationsPossible(["a==b","b!=a"]))
# print(s.equationsPossible(["a==b","b==c","a==c"]))
# print(s.equationsPossible(["a==b","b!=c","c==a"]))
# print(s.equationsPossible(["c==c","b==d","x!=z"]))