https://leetcode-cn.com/problems/satisfiability-of-equality-equations/solution/shou-hui-tu-jie-shou-xie-unionfind-bing-cha-ji-bu-/
不同的字母之间存在联通的关系,因此可以将字母作为图的节点构造成联通图。利用并查集的特性可以快速的判断两个字母之间是否存在连通通路。
路径压缩可以利用树的高度,将高度更高的树的根节点作为两棵树合并后的根节点,也可以更简单地将每个节点都指向根节点,这样查找的时候效率更快。
class Solution {
public boolean equationsPossible(String[] equations) {
UnionFind uf = new UnionFind();
for(int i=0; i<equations.length; i++){
char[] arr = equations[i].toCharArray();
int a = arr[0] - 'a';
int b = arr[3] - 'a';
if(arr[1] == '='){
uf.union(a, b);
}
}
for(int i=0; i<equations.length; i++){
char[] arr = equations[i].toCharArray();
int a = arr[0] - 'a';
int b = arr[3] - 'a';
if(arr[1] == '!' && uf.isConneted(a, b)){
return false;
}
}
return true;
}
class UnionFind{
private int[] parents;
public UnionFind(){
parents = new int[26];
for(int i=0; i<parents.length; i++){
parents[i] = i;
}
}
public void union(int a, int b){
int rootA = find(a);
int rootB = find(b);
parents[rootA] = rootB;
}
public int find(int x){
while(x != parents[x]){
x = parents[parents[x]];
// x = parents[x];
}
return x;
}
public boolean isConneted(int a, int b){
return find(a) == find(b);
}
}
}