题目
给定一个由表示变量之间关系的字符串方程组成的数组,每个字符串方程 equations[i] 的长度为 4,并采用两种不同的形式之一:“a==b” 或 “a!=b”。在这里,a 和 b 是小写字母(不一定不同),表示单字母变量名。
只有当可以将整数分配给变量名,以便满足所有给定的方程时才返回 true,否则返回 false。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/satisfiability-of-equality-equations
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
代码
class Solution {
public boolean equationsPossible(String[] equations) {
UF uf = new UF(26);// 26 个英文字母
// 先让相等的字母形成连通分量
for (String eq:equations
) {
if (eq.charAt(1)=='='){
char x= eq.charAt(0);
char y= eq.charAt(3);
uf.union(x-'a',y-'a');//必须得用确定的int,否则会溢出
}
}
// 检查不等关系是否打破相等关系的连通性
for (String eq:equations
) {
if (eq.charAt(1)=='!'){
char x= eq.charAt(0);
char y= eq.charAt(3);
if (uf.connected(x - 'a', y - 'a'))
return false;
}
// 如果相等关系成立,就是逻辑冲突
}
return true;
}
//并查集
class UF{
// 记录连通分量个数
private int count;
// 存储若干棵树的父节点
private int[] parent;
// 记录树的“重量”
private int[] size;
public UF(int n){
this.count=n;
parent=new int[n];
size=new int[n];
for (int i = 0; i < n; i++) {
parent[i]=i;//每个父节点都是自己
size[i]=1;//每棵树大小都为1
}
}
/* 将 p 和 q 连通 */
public void union(int p,int q){
int rootP=find(p);
int rootQ=find(q);
if (rootP==rootQ) return;//本来就连通
// 小树接到大树下面,较平衡
if (size[rootP]>size[rootQ]){
parent[rootQ]=rootP;
size[rootP]+=size[rootQ];
}else {
parent[rootP]=rootQ;
size[rootQ]+=size[rootP];
}
count--;//连通分量减小
}
/* 判断 p 和 q 是否互相连通 */
public boolean connected(int p, int q){
int rootP=find(p);
int rootQ=find(q);
// 处于同一棵树上的节点,相互连通
return rootP==rootQ;
}
/* 返回节点 x 的根节点 */
private int find(int x){
while (parent[x]!=x){
//顺便进行路径压缩
parent[x]=parent[parent[x]];
x=parent[x];
}
return x;
}
public int count() {//返回连通分量
return count;
}
}
}
要点
- 本题具有「自反性」「传递性」和「对称性」,其实 == 关系也是一种等价关系,具有这些性质,所以这个问题用 Union-Find 算法。
- 先全部判断==的情况,然后判断!=的情况,看看有没有冲突。