[leetcode]207. Course Schedule

 第一种图的方法:超时

struct Edge{
    int to;
    Edge(int x){
        to = x;
    }
};
struct Node{
    vector<Edge> edges;
    int indegree;
    Node(){
        indegree=0;
    }
};
struct Graph{
    Node * node;
    int num;
    Graph(int n){
        node =new Node[n];
        num = n;
    }
    void insert(const pair<int,int> & p){
        node[p.first].edges.push_back(Edge(p.second));
        node[p.second].indegree++;
    }
    ~Graph(){
        delete [] node;
    }
};
class Solution {
public:
    bool canFinish(int numCourses, vector<pair<int, int>>& prerequisites) {
        Graph graph(numCourses);
        for(auto item : prerequisites){
            graph.insert(item);
        }
        bool visited[numCourses];
        queue<int> q;
        for(int i = 0;i < numCourses; i++){
            visited[i]= false;
            if(graph.node[i].indegree == 0){
                q.push(i);
                visited[i]= true;
            }
        }
        while(!q.empty()){
            int index = q.front();
            q.pop();
            for(auto item : graph.node[index].edges){
                graph.node[item.to].indegree--;
            }
            for(int i = 0;i < numCourses; i++){
                if(graph.node[i].indegree == 0){
                    visited[i] = true;
                    q.push(i);
                }
            }
        }
        for(int i = 0;i < numCourses; i++){
            if(!visited[i]){
                return false;
            }
        }
        return true;
        
    }
};

第二种方法:


class Solution {
public:
    bool canFinish(int numCourses, vector<pair<int, int>>& prerequisites) {
        int indegree[numCourses];
        bool visited[numCourses];
        queue<int> q;
        unordered_map<int,vector<int>> m;
        for(int i=0;i<numCourses;i++){
            indegree[i]=0;
            visited[i] = false;
        }
        for(auto item : prerequisites){
            m[item.first].push_back(item.second);
            indegree[item.second]++;
        }
        for(int i=0;i<numCourses;i++){
            if(indegree[i] == 0){
                q.push(i);
                visited[i] = true;     
            }
        }
        while(!q.empty()){
            int index = q.front();
            q.pop();
            // auto p = m.find(index);
            for(auto item : m[index]){
                indegree[item]--;
                if(indegree[item] == 0){
                    q.push(item);
                    visited[item] = true;
                }
            }
        }   
        for(int i=0;i<numCourses;i++){
           if(!visited[i]) 
               return false;
        }
        return true;
    }
};

 思路是拓扑排序,复杂度O(n+e)

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值