Skip to content

Commit 093a37e

Browse files
Sync LeetCode submission Runtime - 4 ms (98.51%), Memory - 65.4 MB (83.51%)
1 parent 0e5dc8f commit 093a37e

2 files changed

Lines changed: 79 additions & 0 deletions

File tree

Lines changed: 52 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,52 @@
1+
<p>There are a total of <code>numCourses</code> courses you have to take, labeled from <code>0</code> to <code>numCourses - 1</code>. You are given an array <code>prerequisites</code> where <code>prerequisites[i] = [a<sub>i</sub>, b<sub>i</sub>]</code> indicates that you <strong>must</strong> take course <code>a<sub>i</sub></code> first if you want to take course <code>b<sub>i</sub></code>.</p>
2+
3+
<ul>
4+
<li>For example, the pair <code>[0, 1]</code> indicates that you have to take course <code>0</code> before you can take course <code>1</code>.</li>
5+
</ul>
6+
7+
<p>Prerequisites can also be <strong>indirect</strong>. If course <code>a</code> is a prerequisite of course <code>b</code>, and course <code>b</code> is a prerequisite of course <code>c</code>, then course <code>a</code> is a prerequisite of course <code>c</code>.</p>
8+
9+
<p>You are also given an array <code>queries</code> where <code>queries[j] = [u<sub>j</sub>, v<sub>j</sub>]</code>. For the <code>j<sup>th</sup></code> query, you should answer whether course <code>u<sub>j</sub></code> is a prerequisite of course <code>v<sub>j</sub></code> or not.</p>
10+
11+
<p>Return <i>a boolean array </i><code>answer</code><i>, where </i><code>answer[j]</code><i> is the answer to the </i><code>j<sup>th</sup></code><i> query.</i></p>
12+
13+
<p>&nbsp;</p>
14+
<p><strong class="example">Example 1:</strong></p>
15+
<img alt="" src="https://assets.leetcode.com/uploads/2021/05/01/courses4-1-graph.jpg" style="width: 222px; height: 62px;" />
16+
<pre>
17+
<strong>Input:</strong> numCourses = 2, prerequisites = [[1,0]], queries = [[0,1],[1,0]]
18+
<strong>Output:</strong> [false,true]
19+
<strong>Explanation:</strong> The pair [1, 0] indicates that you have to take course 1 before you can take course 0.
20+
Course 0 is not a prerequisite of course 1, but the opposite is true.
21+
</pre>
22+
23+
<p><strong class="example">Example 2:</strong></p>
24+
25+
<pre>
26+
<strong>Input:</strong> numCourses = 2, prerequisites = [], queries = [[1,0],[0,1]]
27+
<strong>Output:</strong> [false,false]
28+
<strong>Explanation:</strong> There are no prerequisites, and each course is independent.
29+
</pre>
30+
31+
<p><strong class="example">Example 3:</strong></p>
32+
<img alt="" src="https://assets.leetcode.com/uploads/2021/05/01/courses4-3-graph.jpg" style="width: 222px; height: 222px;" />
33+
<pre>
34+
<strong>Input:</strong> numCourses = 3, prerequisites = [[1,2],[1,0],[2,0]], queries = [[1,0],[1,2]]
35+
<strong>Output:</strong> [true,true]
36+
</pre>
37+
38+
<p>&nbsp;</p>
39+
<p><strong>Constraints:</strong></p>
40+
41+
<ul>
42+
<li><code>2 &lt;= numCourses &lt;= 100</code></li>
43+
<li><code>0 &lt;= prerequisites.length &lt;= (numCourses * (numCourses - 1) / 2)</code></li>
44+
<li><code>prerequisites[i].length == 2</code></li>
45+
<li><code>0 &lt;= a<sub>i</sub>, b<sub>i</sub> &lt;= numCourses - 1</code></li>
46+
<li><code>a<sub>i</sub> != b<sub>i</sub></code></li>
47+
<li>All the pairs <code>[a<sub>i</sub>, b<sub>i</sub>]</code> are <strong>unique</strong>.</li>
48+
<li>The prerequisites graph has no cycles.</li>
49+
<li><code>1 &lt;= queries.length &lt;= 10<sup>4</sup></code></li>
50+
<li><code>0 &lt;= u<sub>i</sub>, v<sub>i</sub> &lt;= numCourses - 1</code></li>
51+
<li><code>u<sub>i</sub> != v<sub>i</sub></code></li>
52+
</ul>
Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,27 @@
1+
class Solution {
2+
public:
3+
vector<bool> checkIfPrerequisite(int numCourses, vector<vector<int>>& prerequisites, vector<vector<int>>& queries) {
4+
vector<vector<int>> graph(numCourses);
5+
for (vector<int>& i : prerequisites) graph[i[0]].push_back(i[1]);
6+
7+
int t = 0;
8+
vector<bitset<100>> isPrereq(numCourses);
9+
vector<bool> seen(numCourses, false);
10+
11+
auto dfs = [&](auto&& self, int node) -> void {
12+
seen[node] = true;
13+
for (int neigh : graph[node]) {
14+
if (!seen[neigh]) self(self, neigh);
15+
isPrereq[node] |= isPrereq[neigh];
16+
isPrereq[node].set(neigh);
17+
}
18+
};
19+
20+
for (int i=0; i<numCourses; i++) if (!seen[i]) dfs(dfs, i);
21+
22+
vector<bool> res;
23+
for (vector<int>& q : queries) res.push_back(isPrereq[q[0]].test(q[1]));
24+
25+
return res;
26+
}
27+
};

0 commit comments

Comments
 (0)