990. 等式方程的可满足性 --- 并查集

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;

}