【leetcode210】dfs/c++

weixin_54470372

已于 2023-02-02 23:24:36 修改

阅读量77

点赞数

文章标签： c++ 宽度优先算法

于 2023-01-28 14:56:35 首次发布

本文链接：https://blog.csdn.net/weixin_54470372/article/details/128709808

版权

leetcode 专栏收录该内容

17 篇文章 0 订阅

订阅专栏

【leetcode210】课程表 II

遍历二维数组（运行超时）

dfs

遍历二维数组（运行超时）

class Solution {
public:
    vector<int> ans;
    vector<bool> visited;
    vector<vector<bool>> graph;
    vector<int> findOrder(int numCourses, vector<vector<int>>& prerequisites) {
        visited.assign(numCourses,0);
        graph.assign(numCourses,visited);
        for(int i=0;i<prerequisites.size();i++){
            graph[prerequisites[i][0]][prerequisites[i][1]]=1;
        }
        for(int i=0;i<numCourses;i++){
            int j;
            if(visited[i])  continue;
            for(j=0;j<numCourses;j++){
                if(graph[i][j] && !visited[j]) break;
            }
            if(j==numCourses){
                ans.push_back(i);
                visited[i]=1;
                i=-1;
            }
        }
        if(ans.size()!=numCourses)  ans.clear();
        return ans;
    }
};

dfs

class Solution {
public:
    vector<int> ans;
    vector<int> visited;
    //0 -- 尚未搜索到
    //1 -- 正在搜索中但未加入队列
    //2 -- 已搜索完毕并加入队列
    vector<vector<bool>> graph;
    bool flag;
    vector<int> findOrder(int numCourses, vector<vector<int>>& prerequisites) {
        visited.assign(numCourses,0);
        graph.assign(numCourses,vector<bool>(numCourses,0));
        flag=0;
        for(int i=0;i<prerequisites.size();i++){
            graph[prerequisites[i][0]][prerequisites[i][1]]=1;
        }
        bfs(numCourses,0);
        for(int i=0;i<numCourses;i++){
            if(visited[i]==0) bfs(numCourses,i);
        }
        if(flag)    ans.clear();
        return ans;
    }

    void bfs(int numCourses,int node){
        visited[node]=1;
        for(int i=0;i<numCourses;i++){
            if(graph[node][i] && visited[i]==0){
                bfs(numCourses,i);
            }
            else if(graph[node][i] && visited[i]==1){
                flag=1;
                break;
            }
        }
        visited[node]=2;
        ans.push_back(node);
    }
};