题目
给定一个由表示变量之间关系的字符串方程组成的数组,每个字符串方程 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
提示:
1 <= equations.length <= 500
equations[i].length == 4
equations[i][0] 和 equations[i][3] 是小写字母
equations[i][1] 要么是 ‘=’,要么是 ‘!’
equations[i][2] 是 ‘=’
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/satisfiability-of-equality-equations
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路
并查集
先扫一遍==并进行合并操作;
再扫一遍!=如果有同一集合则返回false;
返回true。
题解
class Solution {
int find(vector<int>& fa,int x)
{
if(fa[x]!=x){
fa[x]=find(fa,fa[x]);
}
return fa[x];
}
public:
bool equationsPossible(vector<string>& equations) {
if(equations.size()==0) return true;
vector<int> fa(26);
for(int i=0;i<26;i++){
fa[i]=i;
}
int a=0,b=0;
for(int i=0;i<equations.size();i++){
a=find(fa,equations[i][0]-'a');
b=find(fa,equations[i][3]-'a');
if(equations[i][1]=='='){
if(a!=b){
fa[a]=b;
}
}
}
for(int i=0;i<equations.size();i++){
a=find(fa,equations[i][0]-'a');
b=find(fa,equations[i][3]-'a');
if(equations[i][1]=='!'){
if(a==b){
return false;
}
}
}
return true;
}
};