rustgym 0.2.0

rustgym solutions
Documentation 
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
39
40
41
42
43
44

struct Solution;

impl Solution {
    fn eventual_safe_nodes(graph: Vec<Vec<i32>>) -> Vec<i32> {
        let n = graph.len();
        let mut res = vec![];
        let mut state = vec![0; n];
        for i in 0..n {
            if Self::dfs(i, &mut state, &graph) {
                res.push(i as i32);
            }
        }
        res
    }

    fn dfs(u: usize, state: &mut [i32], graph: &[Vec<i32>]) -> bool {
        match state[u] {
            3 => false,
            2 => true,
            1 => {
                state[u] = 3;
                false
            }
            _ => {
                state[u] = 1;
                let mut s = 2;
                for &v in &graph[u] {
                    if !Self::dfs(v as usize, state, graph) {
                        s = 3;
                    }
                }
                state[u] = s;
                state[u] == 2
            }
        }
    }
}

#[test]
fn test() {
    let graph = vec_vec_i32![[1, 2], [2, 3], [5], [0], [5], [], []];
    let res = vec![2, 4, 5, 6];
    assert_eq!(Solution::eventual_safe_nodes(graph), res);
}