题目:力扣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"]))