LeetCode 210 Course Schedule II(BFS / DFS)

题目链接:点击这里

在这里插入图片描述

题意:现在你总共有 n 门课需要选,记为 0 到 n - 1。

在选修某些课程之前需要一些先修课程。 例如,想要学习课程 0 ,你需要先完成课程 1 ,我们用一个匹配来表示他们: [0,1]

给定课程总量以及它们的先决条件,返回你为了学完所有课程所安排的学习顺序。

可能会有多个正确的顺序,你只要返回一种就可以了。如果不可能完成所有课程,返回一个空数组。

方法一:基于BFS

  1. 定义一个队列 Q,并把所有入度为 0 的结点加入队列。
  2. 取队首结点,记录路径,然后删去所有从它出发的边,将这些边到达的顶点的入度减 1,如果某个顶点的入度减为 0 0 0,则将其加入队列。
  3. 反复进行②操作,直到队列为空。

判定:如果队列为空时入过队的结点数目恰好为N,说明拓扑排序成功,图G为有向无环图;否则,拓扑排序失败,图G中有环。

class Solution {
public:

    vector<int> findOrder(int numCourses, vector<vector<int>>& prerequisites) {
        vector<vector<int>> graph(numCourses);  		// 邻接表
        vector<int> inDegree(numCourses);       		// 入度
        vector<int> ans;

        for(auto it : prerequisites)
        {
            graph[it[1]].emplace_back(it[0]);
            inDegree[it[0]]++;
        }

        int num = 0;        						// 记录加入拓扑序列的顶点数
        queue<int> q;

        for(int i = 0; i < inDegree.size(); i++)    // 将所有入度为0的顶点入队
            if(inDegree[i] == 0)
                q.push(i);

        while(!q.empty())
        {
            int u = q.front();  					// 取队首顶点u
            ans.push_back(u);
            q.pop();
            for(int i = 0; i < graph[u].size(); i++)
            {
                int v = graph[u][i];        		// u的后继顶点v
                if(--inDegree[v] == 0)        		// 顶点v的入度减1
                    q.push(v);              		// 顶点v的入度减为0则入队
            }
            num++;
        }
        
        if(num != numCourses) ans.clear();
        
        return ans;
    }
};

方法二:基于DFS

f l a g [   ] flag[\ ] flag[ ] 标记每一个节点的状态, 0 0 0 表示没有访问过, 1 1 1 表示当前的dfs正在访问, − 1 -1 1 表示已经访问完毕。

对所有的节点进行遍历。比如遍历节点 a a a 的时候,遍历前,先将该节点 a a a f l a g flag flag 设置为 1 1 1,表示正在遍历,然后,对 a a a 的邻接表进行遍历:

  1. 如果下一级节点 b b b f l a g flag flag − 1 -1 1,说明已经遍历了,返回 t u r e ture ture
  2. 如果为 1 1 1,说明重复遍历了当前的节点,这个时候存在环,返回 f a l s e false false
  3. 如果为 0 0 0,则继续遍历该节点b的邻接表节点。

DFS 递归返回的首先是最底层的点,它一定是 0 出度点,没有后继点,是拓扑排序的最后一个点;然后逐步回退,最后返回的是起点。因此,是一个逆序。

class Solution {
public:
    stack<int> st;						// 全局变量

    // 若有环,dfs返回false
    bool dfs(int v, vector<vector<int>>& graph, vector<int>& flag)
    {
        if(flag[v] == 1)    return false;
        if(flag[v] == -1)   return true;
        
        flag[v] = 1;
        
        for(auto it : graph[v])
            if(!dfs(it, graph, flag))
                return false;
        
        flag[v] = -1;
        
        st.push(v);
        
        return true;
    }

    vector<int> findOrder(int numCourses, vector<vector<int>>& prerequisites) {
        vector<int> flag(numCourses, 0);        	// 初始化为没有访问过
        vector<vector<int>> graph(numCourses);  	// 邻接表
        vector<int> ans;

        for(auto it : prerequisites)
            graph[it[1]].emplace_back(it[0]);
        
        bool res = true;
        for(int i = 0; i < graph.size(); i++)
            if(!dfs(i, graph, flag))
                res = false;
        
        if(res)
        {
            while(!st.empty())
            {
                ans.push_back(st.top());
                st.pop();
            }
        }
        
        return ans;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值