【LeetCode】2192. 有向无环图中一个节点的所有祖先

原创

distance66 2024-04-28 12:51:32 博主文章分类：数据结构与算法 ©著作权

文章标签 leetcode 数据结构与算法 C++ 文章分类 C/C++ 后端开发

©著作权归作者所有：来自51CTO博客作者distance66的原创作品，请联系作者获取转载授权，否则将追究法律责任

2192. 有向无环图中一个节点的所有祖先

题解

代码

class Solution {
public:

    vector< vector<int> > edge;

    vector<bool> bk;

    vector< set<int> > res;

    set<int> dfs(int cur) {
        if(bk[cur]) {
            return res[cur];
        }
        set<int> st(edge[cur].begin(), edge[cur].end()), st2;
        for(int to : edge[cur]) {
            st2 = dfs(to);
            for(int to2 : st2) {
                st.insert(to2);
            }
        }
        res[cur] = st;
        bk[cur] = true;
        return st;
    }

    vector<vector<int>> getAncestors(int n, vector<vector<int>>& edges) {
        edge = vector< vector<int> > (n, vector<int>());

        for(int i = 0, x, y; i < edges.size(); i++) {
            x = edges[i][0], y = edges[i][1];
            edge[y].push_back(x);
        }

        res = vector< set<int> > (n, set<int>());
        bk = vector<bool> (n, false);

        for(int i = 0; i < n; i++) {
            if(!bk[i]) {
                res[i] = dfs(i);
                bk[i] = true;
            }
        }

        vector< vector<int> > res2 = vector< vector<int> > (n, vector<int>());
        for(int i = 0; i < n; i++) {
            res2[i] = vector<int> (res[i].begin(), res[i].end());
        }
        return res2;
    }
};

题目

给你一个正整数 n ，它表示一个 有向无环图 中节点的数目，节点编号为 0 到 n - 1 （包括两者）。

给你一个二维整数数组 edges ，其中 edges[i] = [from_i, to_i] 表示图中一条从 from_i 到 to_i 的单向边。

请你返回一个数组 answer，其中 answer[i]是第 i 个节点的所有祖先，这些祖先节点升序排序。

如果 u 通过一系列边，能够到达 v ，那么我们称节点 u 是节点 v 的祖先节点。

示例 1：

输入：n = 8, edgeList = [[0,3],[0,4],[1,3],[2,4],[2,7],[3,5],[3,6],[3,7],[4,6]]
输出：[[],[],[],[0,1],[0,2],[0,1,3],[0,1,2,3,4],[0,1,2,3]]
解释：
上图为输入所对应的图。
- 节点 0 ，1 和 2 没有任何祖先。
- 节点 3 有 2 个祖先 0 和 1 。
- 节点 4 有 2 个祖先 0 和 2 。
- 节点 5 有 3 个祖先 0 ，1 和 3 。
- 节点 6 有 5 个祖先 0 ，1 ，2 ，3 和 4 。
- 节点 7 有 4 个祖先 0 ，1 ，2 和 3 。

示例 2：

输入：n = 5, edgeList = [[0,1],[0,2],[0,3],[0,4],[1,2],[1,3],[1,4],[2,3],[2,4],[3,4]]
输出：[[],[0],[0,1],[0,1,2],[0,1,2,3]]
解释：
上图为输入所对应的图。
- 节点 0 没有任何祖先。
- 节点 1 有 1 个祖先 0 。
- 节点 2 有 2 个祖先 0 和 1 。
- 节点 3 有 3 个祖先 0 ，1 和 2 。
- 节点 4 有 4 个祖先 0 ，1 ，2 和 3 。

提示：