Welcome to Subscribe On Youtube

Formatted question description: https://leetcode.ca/all/2050.html

2050. Parallel Courses III (Hard)

You are given an integer n, which indicates that there are n courses labeled from 1 to n. You are also given a 2D integer array relations where relations[j] = [prevCoursej, nextCoursej] denotes that course prevCoursej has to be completed before course nextCoursej (prerequisite relationship). Furthermore, you are given a 0-indexed integer array time where time[i] denotes how many months it takes to complete the (i+1)th course.

You must find the minimum number of months needed to complete all the courses following these rules:

  • You may start taking a course at any time if the prerequisites are met.
  • Any number of courses can be taken at the same time.

Return the minimum number of months needed to complete all the courses.

Note: The test cases are generated such that it is possible to complete every course (i.e., the graph is a directed acyclic graph).

 

Example 1:

Input: n = 3, relations = [[1,3],[2,3]], time = [3,2,5]
Output: 8
Explanation: The figure above represents the given graph and the time required to complete each course. 
We start course 1 and course 2 simultaneously at month 0.
Course 1 takes 3 months and course 2 takes 2 months to complete respectively.
Thus, the earliest time we can start course 3 is at month 3, and the total time required is 3 + 5 = 8 months.

Example 2:

Input: n = 5, relations = [[1,5],[2,5],[3,5],[3,4],[4,5]], time = [1,2,3,4,5]
Output: 12
Explanation: The figure above represents the given graph and the time required to complete each course.
You can start courses 1, 2, and 3 at month 0.
You can complete them after 1, 2, and 3 months respectively.
Course 4 can be taken only after course 3 is completed, i.e., after 3 months. It is completed after 3 + 4 = 7 months.
Course 5 can be taken only after courses 1, 2, 3, and 4 have been completed, i.e., after max(1,2,3,7) = 7 months.
Thus, the minimum time needed to complete all the courses is 7 + 5 = 12 months.

 

Constraints:

  • 1 <= n <= 5 * 104
  • 0 <= relations.length <= min(n * (n - 1) / 2, 5 * 104)
  • relations[j].length == 2
  • 1 <= prevCoursej, nextCoursej <= n
  • prevCoursej != nextCoursej
  • All the pairs [prevCoursej, nextCoursej] are unique.
  • time.length == n
  • 1 <= time[i] <= 104
  • The given graph is a directed acyclic graph.

Similar Questions:

Solution 1. Topological Sort (BFS)

A node’s distance is its time plus the maximum distance of all predecessor nodes. We can calculate the distances via a topological sort. The answer is the maximum distance.

  • // OJ: https://leetcode.com/problems/parallel-courses-iii/
    // Time: O(N + E)
    // Space: O(N + E)
    class Solution {
    public:
        int minimumTime(int n, vector<vector<int>>& E, vector<int>& T) {
            vector<vector<int>> G(n);
            vector<int> indegree(n), dist(n);
            for (auto &e : E) { // build graph and count indegrees
                G[e[0] - 1].push_back(e[1] - 1);
                indegree[e[1] - 1]++;
            }
            queue<int> q;
            for (int i = 0; i < n; ++i) {
                if (indegree[i] == 0) { // enqueue nodes with 0 indegree.
                    q.push(i);
                    dist[i] = T[i]; // source nodes' distance is their corresponding time
                } 
            }
            while (q.size()) {
                int u = q.front();
                q.pop();
                for (int v : G[u]) {
                    dist[v] = max(dist[u] + T[v], dist[v]); // update the distance of node `v` using the maximum distance of predecessor nodes.
                    if (--indegree[v] == 0) q.push(v); // enqueue node `v` when its indegree drops to 0
                }
            }
            return *max_element(begin(dist), end(dist)); // the answer is the maximum distance.
        }
    };
    

    Solution 2. Topological Sort (DFS)

    DFS version topological sort is Post-order Traversal + Memo.

    // OJ: https://leetcode.com/problems/parallel-courses-iii/
    // Time: O(N + E)
    // Space: O(N + E)
    class Solution {
    public:
        int minimumTime(int n, vector<vector<int>>& E, vector<int>& T) {
            vector<vector<int>> G(n);
            vector<int> dist(n);
            for (auto &e : E) G[e[1] - 1].push_back(e[0] - 1);
            function<int(int)> dfs = [&](int u) {
                if (dist[u]) return dist[u];
                int mx = 0;
                for (int v : G[u]) mx = max(mx, dfs(v));
                return dist[u] = mx + T[u];
            };
            for (int i = 0; i < n; ++i) dfs(i);
            return *max_element(begin(dist), end(dist));
        }
    };
    
  • class Solution:
        def minimumTime(self, n: int, relations: List[List[int]], time: List[int]) -> int:
            g = defaultdict(list)
            indeg = [0] * n
            for a, b in relations:
                g[a - 1].append(b - 1)
                indeg[b - 1] += 1
            q = deque()
            dp = [0] * n
            ans = 0
            for i, (v, t) in enumerate(zip(indeg, time)):
                if v == 0:
                    q.append(i)
                    dp[i] = t
                    ans = max(ans, t)
            while q:
                i = q.popleft()
                for j in g[i]:
                    dp[j] = max(dp[j], dp[i] + time[j])
                    ans = max(ans, dp[j])
                    indeg[j] -= 1
                    if indeg[j] == 0:
                        q.append(j)
            return ans
    
    ############
    
    # 2050. Parallel Courses III
    # https://leetcode.com/problems/parallel-courses-iii
    
    class Solution:
        def minimumTime(self, n: int, relations: List[List[int]], time: List[int]) -> int:
            time = [0] + time
            graph = collections.defaultdict(list)
            
            for x, y in relations:
                graph[y].append(x)
                    
            @cache
            def go(x):
                res = 0
                
                for child in graph[x]:
                    res = max(res, go(child))
                
                return res + time[x]
            
            return max(go(i) for i in range(1, n + 1))
                
    
    
  • class Solution {
        public int minimumTime(int n, int[][] relations, int[] time) {
            List<Integer>[] g = new List[n];
            Arrays.setAll(g, k -> new ArrayList<>());
            int[] indeg = new int[n];
            for (int[] e : relations) {
                int a = e[0] - 1, b = e[1] - 1;
                g[a].add(b);
                ++indeg[b];
            }
            Deque<Integer> q = new ArrayDeque<>();
            int[] dp = new int[n];
            int ans = 0;
            for (int i = 0; i < n; ++i) {
                int v = indeg[i], t = time[i];
                if (v == 0) {
                    q.offer(i);
                    dp[i] = t;
                    ans = Math.max(ans, t);
                }
            }
            while (!q.isEmpty()) {
                int i = q.pollFirst();
                for (int j : g[i]) {
                    dp[j] = Math.max(dp[j], dp[i] + time[j]);
                    ans = Math.max(ans, dp[j]);
                    if (--indeg[j] == 0) {
                        q.offer(j);
                    }
                }
            }
            return ans;
        }
    }
    
  • func minimumTime(n int, relations [][]int, time []int) int {
    	g := make([][]int, n)
    	indeg := make([]int, n)
    	for _, e := range relations {
    		a, b := e[0]-1, e[1]-1
    		g[a] = append(g[a], b)
    		indeg[b]++
    	}
    	dp := make([]int, n)
    	q := []int{}
    	ans := 0
    	for i, v := range indeg {
    		if v == 0 {
    			q = append(q, i)
    			dp[i] = time[i]
    			ans = max(ans, time[i])
    		}
    	}
    	for len(q) > 0 {
    		i := q[0]
    		q = q[1:]
    		for _, j := range g[i] {
    			indeg[j]--
    			if indeg[j] == 0 {
    				q = append(q, j)
    			}
    			dp[j] = max(dp[j], dp[i]+time[j])
    			ans = max(ans, dp[j])
    		}
    	}
    	return ans
    }
    
    func max(a, b int) int {
    	if a > b {
    		return a
    	}
    	return b
    }
    
  • function minimumTime(n: number, relations: number[][], time: number[]): number {
        const g: number[][] = Array(n)
            .fill(0)
            .map(() => []);
        const indeg: number[] = Array(n).fill(0);
        for (const [a, b] of relations) {
            g[a - 1].push(b - 1);
            ++indeg[b - 1];
        }
        const q: number[] = [];
        const f: number[] = Array(n).fill(0);
        let ans: number = 0;
        for (let i = 0; i < n; ++i) {
            if (indeg[i] === 0) {
                q.push(i);
                f[i] = time[i];
                ans = Math.max(ans, f[i]);
            }
        }
        while (q.length > 0) {
            const i = q.shift()!;
            for (const j of g[i]) {
                f[j] = Math.max(f[j], f[i] + time[j]);
                ans = Math.max(ans, f[j]);
                if (--indeg[j] === 0) {
                    q.push(j);
                }
            }
        }
        return ans;
    }
    
    

Discuss

https://leetcode.com/problems/parallel-courses-iii/discuss/1537501/C%2B%2B-Topological-Sort

All Problems

All Solutions