2024.4.4力扣每日一题——有向无环图中一个节点的所有祖先

题目来源

力扣每日一题；题序：2192

我的题解

方法一 深度优先遍历 超时

从每一个点开始深度优先遍历。 可以使用访问数组进行优化

时间复杂度：O($n^{2}logn$)
空间复杂度：O($n^2$)

List<Set<Integer>> res;
public List<List<Integer>> getAncestors(int n, int[][] edges) {
    res=new ArrayList<>();
    for(int i=0;i<n;i++){
        res.add(new HashSet<>());
    }
    List<Integer>[] g=createGraph(n,edges);
    for(int i=0;i<n;i++){
        dfs(g,i);
    }
    List<List<Integer>> ans=new ArrayList<>();
    for(Set<Integer> set:res){
        List<Integer> list=new ArrayList<>(set);
        list.sort((a,b)->a-b);
        ans.add(list);
    }
    return ans;
}
public List<Integer>[] createGraph(int n,int[][] edges){
    List<Integer>[] g=new ArrayList[n];
    for(int i=0;i<n;i++){
        g[i]=new ArrayList<>();
    }
    for(int[] t:edges){
        int from = t[0];
        int to = t[1];
        g[from].add(to);
    }
    return g;
}
public void dfs(List<Integer>[] g,int pre){
    for(int next:g[pre]){
        res.get(next).add(pre);
        res.get(next).addAll(res.get(pre));
        dfs(g,next);
    }
}

方法二 拓扑排序

时间复杂度：
空间复杂度：

public List<List<Integer>> getAncestors(int n, int[][] edges) {
	//邻接表
    List<Integer>[] g=new ArrayList[n];
    //入度表
    int[] indeg=new int[n];
    //祖先节点暂存集合
    Set<Integer>[] parent=new HashSet[n];
    //初始化
    for(int i=0;i<n;i++){
        g[i]=new ArrayList<>();
        parent[i]=new HashSet<>();
    }
    for(int[] t:edges){
        int from = t[0];
        int to = t[1];
        g[from].add(to);
        indeg[to]++;
    }
    //优先队列实现拓扑排序
    Queue<Integer> queue=new LinkedList<>();
    for(int i=0;i<n;i++){
        if(indeg[i]==0)
            queue.offer(i);
    }
    while(!queue.isEmpty()){
        int cur=queue.poll();
        for(int next:g[cur]){
       		//当前节点是邻接节点的祖先节点
            parent[next].add(cur);
            //当前节点的祖先节点也是邻接节点的祖先节点
            parent[next].addAll(parent[cur]);
            indeg[next]--;
            if(indeg[next]==0)
                queue.offer(next);
        }
    }
    List<List<Integer>> res=new ArrayList<>();
    for(int i=0;i<n;i++){
        List<Integer> t=new ArrayList<>(parent[i]);
        t.sort((a,b)->a-b);
        res.add(t);
    }
    return res;
}

有任何问题，欢迎评论区交流，欢迎评论区提供其它解题思路（代码），也可以点个赞支持一下作者哈😄~