目录链接:
力扣编程题-解法汇总_分享+记录-CSDN博客
GitHub同步刷题项目:
https://github.com/September26/java-algorithms
原题链接:. - 力扣(LeetCode)
描述:
给你一个正整数 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
- 图中不会有重边。
- 图是 有向 且 无环 的。
解题思路:
我们轻松的梳理出第i个节点依赖哪些节点,本题的难点就是如何按照正确的依赖顺序进行遍历。所以我们可以使用贪心策略,把满足依赖条件的加入到队列中。队列中取出来的就加入到最终的输出集合中。取出的节点如果满足了所有依赖条件,则也加入到队列中。
代码:
class Solution {
public List<List<Integer>> getAncestors(int n, int[][] edges) {
Model2192[] array = new Model2192[n];
for (int i = 0; i < n; i++) {
array[i] = new Model2192(i);
}
for (int i = 0; i < edges.length; i++) {
int from = edges[i][0];
int to = edges[i][1];
array[from].toList.add(to);
array[to].fromSet.add(from);
array[to].dependSet.add(from);
}
List<Integer>[] out = new ArrayList[n];
Queue<Model2192> queue = new ArrayDeque<>();
for (int i = 0; i < n; i++) {
if (array[i].fromSet.size() > 0) {
continue;
}
queue.add(array[i]);
}
while (!queue.isEmpty()) {
Model2192 top = queue.poll();
addToList(out, top);
actionTop(array, queue, top);
}
List<List<Integer>> outList = new ArrayList<>();
for (int i = 0; i < n; i++) {
outList.add(out[i]);
}
return outList;
}
private void actionTop(Model2192[] array, Queue<Model2192> queue, Model2192 poll) {
for (int to : poll.toList) {
array[to].dependSet.remove(poll.value);
array[to].fromSet.addAll(poll.fromSet);
if (array[to].dependSet.size() == 0) {
queue.add(array[to]);
}
}
}
private void addToList(List<Integer>[] out, Model2192 poll) {
List<Integer> list = new ArrayList<>();
out[poll.value] = list;
list.addAll(poll.fromSet);
Collections.sort(list);
}
static class Model2192 {
int value = 0;
List<Integer> toList = new ArrayList<>();
Set<Integer> fromSet = new HashSet<>();
Set<Integer> dependSet = new HashSet<>();
Model2192(int _value) {
this.value = _value;
}
}
}