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
//找父节点的函数(前驱)
int find_father(int *pre,int x)
{
return pre[x] == x ? x : find_father(pre,pre[x]); //如果pre[x]==x说明找到了父根节点,否则继续找父节点的根节点
}
bool equationsPossible(char ** equations, int equationsSize){
int pre[26]; //装入各个字母的前驱关系,下标代表该字母,数组值代表它的前驱(父节点)
for(int i = 0;i < 26;i++) {
pre[i] = i;
}
char uequal[equationsSize][3]; //装入不相等的元素集合
int pos = 0; //不相等元素集合的个数
for(int i = 0; i < equationsSize;i++) {
if(equations[i][1] == '=' && equations[i][0] != equations[i][3]) { //形如 a == b
int pre1 = find_father(pre,equations[i][0] - 'a');
int pre2 = find_father(pre,equations[i][3] - 'a');
pre[pre2] = pre1; //连通关系
}
if(equations[i][1] == '!') {
//如果不等的两个元素符号相同,直接返回false,形如:a != a
if(equations[i][0] == equations[i][3]) {
return false;
}
else { //加入uequal数组中,方便后面检验
uequal[pos][0] = equations[i][0];
uequal[pos++][1] = equations[i][3];
}
}
}
//检查之前不相等的元素集合,如果两个元素的前驱相同,说明违背了不等条件(和相等冲突),则返回false,否则返回true
for(int i = 0;i < pos;i++) {
int pre1 = find_father(pre,uequal[i][0] - 'a');
int pre2 = find_father(pre,uequal[i][1] - 'a');
if(pre1 == pre2)
return false;
}
return true;
}