990. 等式方程的可满足性
给定一个由表示变量之间关系的字符串方程组成的数组,每个字符串方程 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
思路
想到了并查集,可以用并查集处理好==的变量,即是等式两边的变量属于连通分量,通过先处理好等号的变量,然后在遍历不等号的变量,如果不等号两边的变量属于同一连通分量,即可以返回false。
代码
class UnionFind {
private:
vector<int> parent;
public:
UnionFind() {
parent.resize(26);
//开始时并查集内所有节点的根节点均是自己
//iota函数是递增赋值,相当于执行循环parent[i] = i++;
iota(parent.begin(), parent.end(), 0);
}
//查找操作
int find(int index) {
//若根节点是自己,直接返回
if(index == parent[index]) {
return index;
}
//根节点不是自己则继续递归查找,并且赋值,为了路径压缩
parent[index] = find(parent[index]);
return parent[index];
}
//节点联合,通过指定index1的最终根节点为index2的最终根节点即可
void unite(int index1, int index2) {
parent[find(index1)] = find(index2);
}
};
class Solution {
public:
bool equationsPossible(vector<string>& equations) {
UnionFind uf;
//先处理等号
for(const string& str1 : equations) {
if(str1[1] == '=') {
int index1 = str1[0] - 'a';
int index2 = str1[3] - 'a';
uf.unite(index1, index2);
}
}
//处理不等号时,若等号两边变量属于同一连通分量,直接返回false
for(const string& str1 : equations) {
if(str1[1] == '!') {
int index1 = str1[0] - 'a';
int index2 = str1[3] - 'a';
if(uf.find(index1) == uf.find(index2)) {
return false;
}
}
}
return true;
}
};