leetcode 990. 等式方程的可满足性【并查集】

https://leetcode-cn.com/problems/satisfiability-of-equality-equations/solution/shou-hui-tu-jie-shou-xie-unionfind-bing-cha-ji-bu-/
不同的字母之间存在联通的关系,因此可以将字母作为图的节点构造成联通图。利用并查集的特性可以快速的判断两个字母之间是否存在连通通路。
路径压缩可以利用树的高度,将高度更高的树的根节点作为两棵树合并后的根节点,也可以更简单地将每个节点都指向根节点,这样查找的时候效率更快。

class Solution {
    public boolean equationsPossible(String[] equations) {
        UnionFind uf = new UnionFind();
        for(int i=0; i<equations.length; i++){
            char[] arr = equations[i].toCharArray();
            int a = arr[0] - 'a';
            int b = arr[3] - 'a'; 
            if(arr[1] == '='){
                uf.union(a, b);
            }
        }

        for(int i=0; i<equations.length; i++){
            char[] arr = equations[i].toCharArray();
            int a = arr[0] - 'a';
            int b = arr[3] - 'a'; 
            if(arr[1] == '!' && uf.isConneted(a, b)){
                return false;
            }
        }
        return true;
    }

    class UnionFind{
        private int[] parents;
        public UnionFind(){
            parents = new int[26];
            for(int i=0; i<parents.length; i++){
                parents[i] = i;
            }
        }

        public void union(int a, int b){
            int rootA = find(a);
            int rootB = find(b);
            parents[rootA] = rootB;
        }

        public int find(int x){
            while(x != parents[x]){
                x = parents[parents[x]];
                // x = parents[x];
            }
            return x;
        }

        public boolean isConneted(int a, int b){
            return find(a) == find(b);
        }
    }
}

你可能感兴趣的:(刷题路漫漫)