LeetCode 207:课程表(拓扑排序查DAG)

LeetCode 207:课程表(拓扑排序查DAG)
题目:

你这个学期必须选修 numCourse 门课程,记为 0 到 numCourse-1 。

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

给定课程总量以及它们的先决条件,请你判断是否可能完成所有课程的学习?

示例 1:

输入: 2, [[1,0]]
输出: true
解释: 总共有 2 门课程。学习课程 1 之前,你需要完成课程 0。所以这是可能的。
示例 2:

输入: 2, [[1,0],[0,1]]
输出: false
解释: 总共有 2 门课程。学习课程 1 之前,你需要先完成​课程 0;并且学习课程 0 之前,你还应先完成课程 1。这是不可能的。

思路:

很典型的判断有向图有环问题。

自己的答案采用了DFS做,但是递归超时,看看怎么改进

用拓扑排序的入度判断更快

class Solution {
public:
    bool canFinish(int numCourses, vector<vector<int>>& prerequisites) {
        vector<vector<int>> Graph(numCourses,vector<int>());
        vector<int> indegree(numCourses,0);
        for(int i=0;i<prerequisites.size();i++){
            indegree[prerequisites[i][1]]++;
            Graph[prerequisites[i][0]].push_back(prerequisites[i][1]);
        }
        // 度为0的点入队
        queue<int> myqueue;
        for(int i=0;i<numCourses;i++){
            if(indegree[i]==0) myqueue.push(i);
        }
        int cnt=0;
        while(!myqueue.empty()){
            int temp = myqueue.front();
            myqueue.pop();
            cnt++; // 有元素出队就加一次
            for(int i=0;i<Graph[temp].size();i++){
                indegree[Graph[temp][i]]--;
                if(indegree[Graph[temp][i]]==0)
                    myqueue.push(Graph[temp][i]);
            }
        }
        return cnt==numCourses;
    }
};
自己的答案(超时)
class Solution {
public:
    bool canFinish(int numCourses, vector<vector<int>>& prerequisites) {
        if(numCourses<2) return true;
        // 思路:判断环
        // step1: 存图
        vector<vector<bool>> Graph(numCourses,vector<bool>(numCourses,false));
        for(int i=0;i<prerequisites.size();i++){
            Graph[prerequisites[i][0]][prerequisites[i][1]] = true;
        }
        // step2: DFS遍历
        vector<bool> visited(numCourses,false);
        for(int i=0;i<numCourses;i++){
            if(visited[i]==false){
                set<int> s;
                if(DFS(i,Graph,visited,s)==false) return false;
            }
        }
        return true;
    }
    
    bool DFS(int start, vector<vector<bool>> Graph, vector<bool>& visited,set<int> &s){
        visited[start]=true;
        s.insert(start);
        for(int i=0;i<visited.size();i++){
            if(Graph[start][i]==true){
                if(s.find(i)==s.end()){ 
                    if(DFS(i,Graph,visited,s)==false) return false;
                }else return false;
            }
        }
        s.erase(start);
        return true;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值