leetcode990. 等式方程的可满足性
题目描述
链接: leetcode990.
给定一个由表示变量之间关系的字符串方程组成的数组,每个字符串方程 equations[i] 的长度为 4,并采用两种不同的形式之一:“a==b” 或"a!=b"。在这里,a 和 b 是小写字母(不一定不同),表示单字母变量名。
只有当可以将整数分配给变量名,以便满足所有给定的方程时才返回true,否则返回 false。
示例1:
输入:["a==b","b!=a"]
输出:false
解释:如果我们指定,a = 1 且 b = 1,那么可以满足第一个方程,但无法满足第二个方程。没有办法分配变量同时满足这两个方程。
示例2:
输入:["b==a","a==b"]
输出:true
解释:我们可以指定 a = 1 且 b = 1 以满足满足这两个方程。
示例3:
输入:["a==b","b==c","a==c"]
输出:true
示例4:
输入:["a==b","b!=c","c==a"]
输出:false
示例5:
输入:["c==c","b==d","x!=z"]
输出:true
题解
使用并查集来解答:
- 初始化并查集,大小为26
- 遍历数组,若符号为 == ,则表示该两个字母连通,使用union方法连接,形成连通分量
- 再次遍历数组,对于符号为 != 的方程式,判断左右两边的变量是否连通,若连通,则产生矛盾,直接返回false
其中并查集算法可见: 并查集算法.
public boolean equationsPossible(String[] equations) {
UnionFind unionFind = new UnionFind(26);
for (String str : equations) {
if (str.contains("==")) {
char x = str.charAt(0);
char y = str.charAt(3);
unionFind.union(x - 'a', y - 'a');
}
}
for (String str : equations) {
if (str.contains("!=")) {
char x = str.charAt(0);
char y = str.charAt(3);
if (unionFind.connected(x - 'a', y - 'a')) {
return false;
}
}
}
return true;
}