leetcode 990 等式方程 (6-8刷題)

方法:並查集。

思路:相等的關係可以傳遞,最終的結果集必然是若干個集合,每個集合裏的變量擁有相同的值,即處在同一個連通分量裏。而不等關係則是兩個變量在不同的連通分量裏。
所以,先針對相等關係建立並查集,而後判斷每個不等關係是否都滿足即可。
代碼:

class Solution {
    // 並查集,這裏並查集的根節點與下標相等。
    public int[] parent=new int[26];
    public int findParent(int t){
        while(parent[t]!=t){
            t=parent[t];
        }
        return t;
    }
    public void union(int a,int b){
        int pa=findParent(a),pb=findParent(b);
        if(pa==pb)return;
        parent[pb]=pa;
    }
    public boolean equationsPossible(String[] equations) {
        // 並查集初始化。
        for(int i=0;i<parent.length;i++){
            parent[i]=i;
        }
        for(int i=0;i<equations.length;i++){
            if(equations[i].charAt(1)=='='){
              union(equations[i].charAt(0)-'a',equations[i].charAt(3)-'a');
            }
        }
        for(int i=0;i<equations.length;i++){
            if(equations[i].charAt(1)=='!'){
                if(findParent(equations[i].charAt(0)-'a')==findParent(equations[i].charAt(3)-'a')){
                    return false;
                }
            }
        }
        return true;
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章