Medium

Course Schedule IIC++

Full explanation · Time O(\ · Space V\

// Time:  O(|V| + |E|)
// Space: O(|E|)

// Kahn’s algorithm (bfs solution)
class Solution {
public:
    vector<int> findOrder(int numCourses, vector<vector<int>>& prerequisites) {
        unordered_map<int, vector<int>> adj;
        unordered_map<int, int> in_degree;
        for (const auto& p : prerequisites) {
            ++in_degree[p[0]];
            adj[p[1]].emplace_back(p[0]);
        }
        queue<int> q;
        for (int i = 0; i < numCourses; ++i) {
            if (!in_degree.count(i)) {
                q.emplace(i);
            }
        }
        vector<int> result;
        while (!q.empty()) {
            const auto node = q.front(); q.pop();
            result.emplace_back(node);
            for (const auto& i : adj[node]) {
                if (!--in_degree[i]) {
                    q.emplace(i);
                }
            }
        }
        return size(result) == numCourses ? result : vector<int>();
    }
};

// Time:  O(|V| + |E|)
// Space: O(|E|)
// dfs solution
class Solution2 {
public:
    vector<int> findOrder(int numCourses, vector<vector<int>>& prerequisites) {
        unordered_map<int, vector<int>> adj;
        unordered_map<int, int> in_degree;
        for (const auto& p : prerequisites) {
            ++in_degree[p[0]];
            adj[p[1]].emplace_back(p[0]);
        }
        vector<int> stk;
        for (int i = 0; i < numCourses; ++i) {
            if (!in_degree.count(i)) {
                stk.emplace_back(i);
            }
        }
        vector<int> result;
        while (!stk.empty()) {
            const auto node = stk.back(); stk.pop_back();
            result.emplace_back(node);
            for (const auto& i : adj[node]) {
                if (!--in_degree[i]) {
                    stk.emplace_back(i);
                }
            }
        }
        return size(result) == numCourses ? result : vector<int>();
    }
};