拓扑排序:深度优先搜索实现(leetcode210课程表)

1. 概念

拓扑排序:对于n个节点的有向图G,一个有向边(u, v),那么对节点进行排序后,u在v的前面(如果G中有环存在,那么不能进行拓扑排序)。
可以用dfs(栈)和bfs(队列)来实现拓扑排序。

dfs具体算法:将所有的点设为三种状态:未搜索0,搜索中1,已完成2。对任意一个未搜索节点0的相邻节点进行搜索,同时把这个节点的状态设为1,1的相邻节点只能是0和2,如果状态为1的节点的相邻节点状态为1,说明有环,返回null。如果没有相邻节点,保存到栈中,把这个节点的状态设为2。

2. 代码

class Solution {
    // 存储有向图
    List<List<Integer>> map;
    // 用一个数组来模拟栈
    int[] result;
    int index;
    // 保存每一个节点的访问状态
    int[] visited;
    // 如果有环存在,valid=false
    boolean valid;

    public int[] findOrder(int numCourses, int[][] prerequisites) {
        map = new ArrayList<>();
        result = new int[numCourses];
        index = numCourses;
        visited = new int[numCourses];
        valid = true;

        for(int i = 0; i < numCourses; i++){
            map.add(new ArrayList<>());
        }
        for(int[] edge : prerequisites){
            //构建有向图
            map.get(edge[1]).add(edge[0]);
        }
        for(int i = 0; i < numCourses && valid; i++){
            if(visited[i] == 0){
                dfs(i);
            }
        }
        //说明有环
        if(!valid){return new int[]{};}
        return result;
    }
    public void dfs(int u){
        visited[u] = 1;
        for(int v : map.get(u)){
            if(visited[v] == 1){
                valid = false;
                return;
            }else if(visited[v] == 0){
                dfs(v);
            }
        }
        visited[u] = 2;
        //存储到栈中
        result[--index] = u;
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

祖国の花朵

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值