erlang算法系列-leetclode-2192. 有向无环图中一个节点的所有祖先(中等)

给你一个正整数 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
图中不会有重边。
图是 有向 且 无环 的。

解题思路

拓扑排序,使用map存贮。每增加一次edges[i] = [fromi, toi]时,把fromi所有祖先节点加入到toi的后续节点,注意去重。

-spec get_ancestors(N :: integer(), Edges :: [[integer()]]) -> [[integer()]].
get_ancestors(N, Edges) ->
    Seq = lists:seq(1, N),
    Map1 = maps:new(),
    Map2 = maps:new(),
    {Map11, Map22}=lists:foldl(fun(I, {M1, M2}) ->
        {maps:put(I-1, [], M1), maps:put(I-1, [], M2)}
    end, {Map1, Map2}, Seq),
    do(Edges, Map11, Map22)
  .
  
 do([], M1, M2) ->
    List = lists:sort(maps:to_list(M2)),
    [lists:sort(L) || {_K, L} <- List];
do([[I,J] | L], M1, M2) ->
    L1 = maps:get(I, M2),
    M21 = find([J], [I | L1],  [], M1, M2),
    F = maps:get(I, M1) -- [J],
    M11 = maps:update(I, [J | F], M1),
    do(L, M11, M21).
    
find([], L, K,M1, M2) ->
    M2;
find([J | J1], L, K, M1, M2) ->
    F = maps:get(J, M1) -- K,
    L1 = maps:get(J, M2),
    L2 = L -- L1, 
    M21 = maps:update(J, L2 ++ L1, M2),
    if
        F =:= [] ->
            find(F ++ J1, L, [J | K], M1, M21);
        true ->
            find(F ++ J1, L, [J|K],M1, M21)
    end.
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值