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
45
46
47
48
49
50
51
52
53

struct Solution;
use std::cmp::Ordering::*;

impl Solution {
    fn collide(stack: &mut Vec<i32>, asteroid: i32) {
        if let Some(&top) = stack.last() {
            if top < 0 {
                stack.push(asteroid);
            } else {
                match top.cmp(&-asteroid) {
                    Less => {
                        stack.pop();
                        Self::collide(stack, asteroid);
                    }
                    Equal => {
                        stack.pop();
                    }
                    Greater => {}
                }
            }
        } else {
            stack.push(asteroid);
        }
    }

    fn asteroid_collision(asteroids: Vec<i32>) -> Vec<i32> {
        let mut res: Vec<i32> = vec![];
        for x in asteroids {
            if x > 0 {
                res.push(x);
            } else {
                Self::collide(&mut res, x);
            }
        }
        res
    }
}

#[test]
fn test() {
    let asteroids: Vec<i32> = vec![5, 10, -5];
    let res = vec![5, 10];
    assert_eq!(Solution::asteroid_collision(asteroids), res);
    let asteroids: Vec<i32> = vec![8, -8];
    let res = vec![];
    assert_eq!(Solution::asteroid_collision(asteroids), res);
    let asteroids: Vec<i32> = vec![10, 2, -5];
    let res = vec![10];
    assert_eq!(Solution::asteroid_collision(asteroids), res);
    let asteroids: Vec<i32> = vec![-2, -1, 1, 2];
    let res = vec![-2, -1, 1, 2];
    assert_eq!(Solution::asteroid_collision(asteroids), res);
}