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

use std::collections::HashSet;
use std::iter::FromIterator;

struct PhoneDirectory {
    available: HashSet<i32>,
}

impl PhoneDirectory {
    fn new(max_numbers: i32) -> Self {
        let available = HashSet::from_iter(0..max_numbers);
        PhoneDirectory { available }
    }

    fn get(&mut self) -> i32 {
        if let Some(&x) = self.available.iter().next() {
            self.available.remove(&x);
            x
        } else {
            -1
        }
    }

    fn check(&self, number: i32) -> bool {
        self.available.contains(&number)
    }

    fn release(&mut self, number: i32) {
        self.available.insert(number);
    }
}

#[test]
fn test() {
    let mut obj = PhoneDirectory::new(1);
    assert_eq!(obj.get(), 0);
    assert_eq!(obj.check(0), false);
    assert_eq!(obj.get(), -1);
    obj.release(2);
    assert_eq!(obj.check(2), true);
}