leetcode-990. 等式方程的可满足性-C语言

最新推荐文章于 2024-01-03 20:29:32 发布

原创最新推荐文章于 2024-01-03 20:29:32 发布 · 560 阅读

2 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

251 篇文章

订阅专栏

本文介绍了一种使用并查集算法解决方程组问题的方法。通过将相等的元素合并到同一集合中，并检查不等式两端元素是否位于相同集合来判断方程组的可行性。该算法首先遍历所有等于号表达式，合并相关元素，然后检查所有不等于号表达式，确保其两端元素不属于同一集合。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

/*
 * 算法思想：并查集
 * 1. 将相等的元素添加到一个并查集中；
 * 2. 将不等的式子，看两端元素是否在前述并查集中，如果在则返回false。
 */

int find(int *arr, int i){
    return i == arr[i] ? i : (arr[i] = find(arr, arr[i]));
}

void un(int *arr, int i, int j){
    int x = find(arr, i);
    int y = find(arr, j);
    
    arr[y] = x;
}

bool equationsPossible(char ** eq, int eqsize){
    int i, j, arr[128];
    
    for(i=0; i<128; i++){
        arr[i] = i;
    }
    
    /* 将相等的元素添加到一个集合中， */
    for(i=0; i<eqsize; i++){
        if(eq[i][1] == '=') {
            un(arr, (int)eq[i][0], (int)eq[i][3]);
        }    
    }
    
    /* 查找不等的元素，是否在相等的集合中 */
    for(i=0; i<eqsize; i++){
        if(eq[i][1] == '!' && find(arr, (int)eq[i][0]) == find(arr, (int)eq[i][3])) {
            return false;
        }    
    }
    
    return true;
}