LeetCode 2192. 有向无环图中一个节点的所有祖先（拓扑排序）

文章目录

- 1. 题目
- 2. 解题

1. 题目

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

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

请你返回一个数组 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 。提示：
1 <= n <= 1000
0 <= edges.length <= min(2000, n * (n - 1) / 2)
edges[i].length == 2
0 <= fromi, toi <= n - 1
fromi != toi
图中不会有重边。
图是 有向 且 无环 的。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/all-ancestors-of-a-node-in-a-directed-acyclic-graph
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

拓扑排序的思路
建图，入度计数，从入度为 0 的开始遍历

class Solution {
public:vector<vector<int>> getAncestors(int n, vector<vector<int>>& edges) {vector<set<int>> ancest(n);vector<int> indegree(n);vector<unordered_set<int>> g(n);for(auto& e : edges){g[e[0]].insert(e[1]);//建图indegree[e[1]]++;//入度}queue<int> q;for(int i = 0; i < n; ++i){if(indegree[i] == 0)q.push(i);//入度为0加入队列}while(!q.empty()){int id = q.front();q.pop();for(auto nid : g[id]){for(auto a : ancest[id]){ancest[nid].insert(a);//继承之前节点的祖先}ancest[nid].insert(id);//加入from节点为祖先if(--indegree[nid] == 0)q.push(nid);}}vector<vector<int>> ans(n);for(int i = 0; i < n; ++i){ans[i] = vector<int>(ancest[i].begin(), ancest[i].end());//转化为答案}return ans;}
};