leetcode 207. Course Schedule(拓扑排序)

来源:互联网 发布:看杀卫玠知乎 编辑:程序博客网 时间:2024/06/05 10:10

题意:

给你一些课程的学习先后顺序,问是否会发生矛盾。

思路:

判断是否产生有向环。并查集只能判断无向环,拓扑排序可判断有向环。

代码:

class Solution {public:    bool canFinish(int numCourses, vector<pair<int, int>>& prerequisites) {        int n = prerequisites.size();        int c[numCourses];        memset(c,0,sizeof(c));        vector<vector<int> > arr(numCourses);        for(int i = 0;i<n;i++)        {            int x = prerequisites[i].first;            int y = prerequisites[i].second;            arr[x].push_back(y);        }        for(int i = 0;i<numCourses;i++)        {            if(c[i]==0)            {                if(!dfs(i,c,arr))                    return false;            }        }        return true;    }    bool dfs(int u,int* c,vector<vector<int>> &arr)    {        c[u] = -1;        for(int i = 0;i<arr[u].size();i++)        {            int v = arr[u][i];            if(c[v]<0)                return false;            else if(c[v]==0&&!dfs(v,c,arr))                return false;        }        c[u] = 1;        return true;    }};

原创粉丝点击