Java 支配樹 DAG上LCA

https://www.zhihu.com/question/46440863?sort=created
https://blog.csdn.net/qq_41955236/article/details/98472911
https://www.luogu.com.cn/problem/P2597

import java.util.*;

public class DagLCA {

    int nodeCnt;
    int[] topSort;
    int[] inDegrees;
    int[][] st; // 父節(jié)點(diǎn)跳表
    Map<Integer, GraphNode> nodeMap;
    List<Integer> roots;

    Map<Integer, GraphNode> dominatorTree;
    int[] parent; // 支配樹上的父節(jié)點(diǎn)
    int[] depths; // 支配樹上的深度
    private static final int ABSOLUTE_ROOT_ID = 0;
    private int dominatorTreeRootId;


    public DagLCA(int[][] edges, int nodeCnt) {
        this.nodeCnt = nodeCnt;
        topSort = new int[nodeCnt + 1];
        inDegrees = new int[nodeCnt + 1];
        parent = new int[nodeCnt + 1];
        depths = new int[nodeCnt + 1];
        nodeMap = new HashMap<>();
        dominatorTree = new HashMap<>();
        st = new int[nodeCnt + 1][nodeCnt];
        Arrays.fill(parent, -1);
        parent[0] = 0;
        for (int[] edge : edges) {
            inDegrees[edge[1]]++;
            addEdge(edge[0], edge[1], nodeMap);
        }
        topologySort();
        System.out.println(dominatorTree);
    }

    private void addEdge(int from, int to, Map<Integer, GraphNode> graph) {
        GraphNode fromNode = initOrGetGraphNode(from, graph);
        GraphNode toNode = initOrGetGraphNode(to, graph);
        fromNode.next.add(to);
        toNode.pre.add(from);
    }

    private GraphNode initOrGetGraphNode(int id, Map<Integer, GraphNode> graph) {
        GraphNode node;
        if (!graph.containsKey(id)) {
            node = new GraphNode(id);
            graph.put(id, node);
        }
        return graph.get(id);
    }

    private void topologySort() {
        roots = getDAGGraphRoot();
        Deque<Integer> nextUpdateNodes = new LinkedList<>();

        if (roots.size() > 1) {
            addAbsoluteRoot();
            dominatorTreeRootId = ABSOLUTE_ROOT_ID;
            nextUpdateNodes.add(ABSOLUTE_ROOT_ID);
        } else {
            dominatorTreeRootId = roots.get(0);
            parent[dominatorTreeRootId] = dominatorTreeRootId;
            st[dominatorTreeRootId][0] = dominatorTreeRootId;
            nextUpdateNodes.addAll(roots);
        }

        while (!nextUpdateNodes.isEmpty()) {
            int nodeId = nextUpdateNodes.pop();
            GraphNode DAGNode = nodeMap.get(nodeId);

            // 連接自身和父節(jié)點(diǎn),當(dāng)走到這一步時(shí),parent數(shù)組一定已經(jīng)完成更新
            if (nodeId != dominatorTreeRootId) {
                addEdge(parent[nodeId], nodeId, dominatorTree);
                st[nodeId][0] = parent[nodeId];
                depths[nodeId] = depths[parent[nodeId]] + 1;
            }

            // 更新父節(jié)點(diǎn)跳表
            for (int i = 1; i < nodeCnt; i++)
                st[nodeId][i] = st[st[nodeId][i - 1]][i - 1];

            // 遍歷子節(jié)點(diǎn)疼鸟,更新入度并求前驅(qū)LCA
            for (int child : DAGNode.next) {
                // 如果子節(jié)點(diǎn)前驅(qū)只有一個(gè)則被唯一前驅(qū)支配放棒,
                // 如果有多個(gè)則被所有前驅(qū)的最近公共父節(jié)點(diǎn)支配搬卒。
                if (nodeMap.get(child).pre.size() == 1) {
                    parent[child] = nodeId;
                } else if (parent[child] != -1) {
                    parent[child] = lca(parent[child], nodeId);
                } else if (parent[child] == -1) {
                    parent[child] = nodeId;
                }
                // 子節(jié)點(diǎn)入度減1
                inDegrees[child]--;
                // 入度為0的節(jié)點(diǎn)入隊(duì)
                if (inDegrees[child] == 0) nextUpdateNodes.add(child);
            }

        }
    }

    private List<Integer> getDAGGraphRoot() {
        List<Integer> graphRootNodes = new ArrayList<>();
        for (int i = 1; i < inDegrees.length; i++) {
            if (inDegrees[i] == 0) graphRootNodes.add(i);
        }
        return graphRootNodes;
    }

    // 如果同時(shí)有多個(gè)起點(diǎn)宇植,需要為整個(gè)圖配置一個(gè)唯一的根節(jié)點(diǎn)潜秋。
    private void addAbsoluteRoot() {
        initOrGetGraphNode(0, nodeMap);
        initOrGetGraphNode(0, dominatorTree);
        for (int rootId : roots) {
            parent[rootId] = ABSOLUTE_ROOT_ID;
            addEdge(0, rootId, dominatorTree);
            addEdge(0, rootId, nodeMap);
            depths[rootId]++;
            inDegrees[rootId]++;
        }
    }

    public int lca(int x, int y) {
        if (x == y) return x;
        if (depths[x] < depths[y]) {
            int tmp = x;
            x = y;
            y = tmp;
        }
        for (int i = nodeCnt - 1; i >= 0; i--)
            if (depths[st[x][i]] >= depths[y])
                x = st[x][i];
        if (x == y) return x;
        for (int i = nodeCnt - 1; i >= 0; i--) {
            if (st[x][i] != st[y][i]) {
                x = st[x][i];
                y = st[y][i];
            }
        }
        return st[x][0];
    }


    public static void main(String[] args) {
        int[][] edges = new int[][]{{1, 3}, {1, 4}, {2, 4}, {2, 5}, {3, 6}, {4, 6}, {5, 6}, {5, 7}, {6, 8}, {6, 9}, {8, 11}, {8, 10}, {9, 10}, {9, 12}, {7, 13}, {13, 14}, {14,15}};
        DagLCA lca = new DagLCA(edges, 15);
        System.out.println(lca.lca(10, 11));
    }

}
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個(gè)濱河市符衔,隨后出現(xiàn)的幾起案子找前,更是在濱河造成了極大的恐慌,老刑警劉巖判族,帶你破解...
    沈念sama閱讀 216,544評(píng)論 6 501
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件躺盛,死亡現(xiàn)場(chǎng)離奇詭異,居然都是意外死亡形帮,警方通過(guò)查閱死者的電腦和手機(jī)槽惫,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 92,430評(píng)論 3 392
  • 文/潘曉璐 我一進(jìn)店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來(lái)辩撑,“玉大人躯枢,你說(shuō)我怎么就攤上這事』蓖危” “怎么了锄蹂?”我有些...
    開封第一講書人閱讀 162,764評(píng)論 0 353
  • 文/不壞的土叔 我叫張陵,是天一觀的道長(zhǎng)水慨。 經(jīng)常有香客問(wèn)我得糜,道長(zhǎng),這世上最難降的妖魔是什么晰洒? 我笑而不...
    開封第一講書人閱讀 58,193評(píng)論 1 292
  • 正文 為了忘掉前任朝抖,我火速辦了婚禮,結(jié)果婚禮上谍珊,老公的妹妹穿的比我還像新娘治宣。我一直安慰自己,他們只是感情好砌滞,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,216評(píng)論 6 388
  • 文/花漫 我一把揭開白布侮邀。 她就那樣靜靜地躺著,像睡著了一般贝润。 火紅的嫁衣襯著肌膚如雪绊茧。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 51,182評(píng)論 1 299
  • 那天打掘,我揣著相機(jī)與錄音华畏,去河邊找鬼。 笑死尊蚁,一個(gè)胖子當(dāng)著我的面吹牛亡笑,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播横朋,決...
    沈念sama閱讀 40,063評(píng)論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼仑乌,長(zhǎng)吁一口氣:“原來(lái)是場(chǎng)噩夢(mèng)啊……” “哼!你這毒婦竟也來(lái)了?” 一聲冷哼從身側(cè)響起绝骚,我...
    開封第一講書人閱讀 38,917評(píng)論 0 274
  • 序言:老撾萬(wàn)榮一對(duì)情侶失蹤耐版,失蹤者是張志新(化名)和其女友劉穎,沒想到半個(gè)月后压汪,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體粪牲,經(jīng)...
    沈念sama閱讀 45,329評(píng)論 1 310
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,543評(píng)論 2 332
  • 正文 我和宋清朗相戀三年止剖,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了腺阳。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點(diǎn)故事閱讀 39,722評(píng)論 1 348
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡穿香,死狀恐怖亭引,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情皮获,我是刑警寧澤焙蚓,帶...
    沈念sama閱讀 35,425評(píng)論 5 343
  • 正文 年R本政府宣布,位于F島的核電站洒宝,受9級(jí)特大地震影響购公,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜雁歌,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,019評(píng)論 3 326
  • 文/蒙蒙 一宏浩、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧靠瞎,春花似錦比庄、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,671評(píng)論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)。三九已至丑勤,卻和暖如春华嘹,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背法竞。 一陣腳步聲響...
    開封第一講書人閱讀 32,825評(píng)論 1 269
  • 我被黑心中介騙來(lái)泰國(guó)打工, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留强挫,地道東北人岔霸。 一個(gè)月前我還...
    沈念sama閱讀 47,729評(píng)論 2 368
  • 正文 我出身青樓,卻偏偏與公主長(zhǎng)得像俯渤,于是被迫代替她去往敵國(guó)和親呆细。 傳聞我的和親對(duì)象是個(gè)殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 44,614評(píng)論 2 353

推薦閱讀更多精彩內(nèi)容