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; // 父节点跳表
    Map<Integer, GraphNode> nodeMap;
    List<Integer> roots;

    Map<Integer, GraphNode> dominatorTree;
    int[] parent; // 支配树上的父节点
    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);

            // 连接自身和父节点,当走到这一步时,parent数组一定已经完成更新
            if (nodeId != dominatorTreeRootId) {
                addEdge(parent[nodeId], nodeId, dominatorTree);
                st[nodeId][0] = parent[nodeId];
                depths[nodeId] = depths[parent[nodeId]] + 1;
            }

            // 更新父节点跳表
            for (int i = 1; i < nodeCnt; i++)
                st[nodeId][i] = st[st[nodeId][i - 1]][i - 1];

            // 遍历子节点,更新入度并求前驱LCA
            for (int child : DAGNode.next) {
                // 如果子节点前驱只有一个则被唯一前驱支配,
                // 如果有多个则被所有前驱的最近公共父节点支配。
                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;
                }
                // 子节点入度减1
                inDegrees[child]--;
                // 入度为0的节点入队
                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;
    }

    // 如果同时有多个起点,需要为整个图配置一个唯一的根节点。
    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));
    }

}
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 203,772评论 6 477
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,458评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 150,610评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,640评论 1 276
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,657评论 5 365
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,590评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,962评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,631评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,870评论 1 297
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,611评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,704评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,386评论 4 319
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,969评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,944评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,179评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 44,742评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,440评论 2 342

推荐阅读更多精彩内容