-
-
Notifications
You must be signed in to change notification settings - Fork 8.8k
/
Copy pathSolution.rs
38 lines (38 loc) · 1.21 KB
/
Solution.rs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
impl Solution {
pub fn find_order(num_courses: i32, prerequisites: Vec<Vec<i32>>) -> Vec<i32> {
let n = num_courses as usize;
let mut adjacency = vec![vec![]; n];
let mut entry = vec![0; n];
// init
for iter in prerequisites.iter() {
let (a, b) = (iter[0], iter[1]);
adjacency[b as usize].push(a);
entry[a as usize] += 1;
}
// construct deque & reslut
let mut deque = std::collections::VecDeque::new();
for index in 0..n {
if entry[index] == 0 {
deque.push_back(index);
}
}
let mut result = vec![];
// bfs
while !deque.is_empty() {
let head = deque.pop_front().unwrap();
result.push(head as i32);
// update degree of entry
for &out_entry in adjacency[head].iter() {
entry[out_entry as usize] -= 1;
if entry[out_entry as usize] == 0 {
deque.push_back(out_entry as usize);
}
}
}
if result.len() == n {
result
} else {
vec![]
}
}
}