前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >LeetCode 5300. 有向无环图中一个节点的所有祖先(拓扑排序)

LeetCode 5300. 有向无环图中一个节点的所有祖先(拓扑排序)

作者头像
Michael阿明
发布2022-03-10 18:32:42
8210
发布2022-03-10 18:32:42
举报
文章被收录于专栏:Michael阿明学习之路

文章目录

1. 题目

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

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

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

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

示例 1:

代码语言:javascript
复制
输入: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:

代码语言:javascript
复制
输入: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 的开始遍历
代码语言:javascript
复制
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;
    }
};

492 ms 140.6 MB C++


我的CSDN博客地址 https://michael.blog.csdn.net/

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 文章目录
  • 1. 题目
  • 2. 解题
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档