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

use std::collections::VecDeque;

#[derive(Default)]
struct RecentCounter {
    queue: VecDeque<i32>,
}

impl RecentCounter {
    fn new() -> Self {
        RecentCounter {
            queue: VecDeque::new(),
        }
    }
    fn ping(&mut self, t: i32) -> i32 {
        self.queue.push_back(t);
        while let Some(p) = self.queue.pop_front() {
            if p >= t - 3000 {
                self.queue.push_front(p);
                break;
            }
        }
        self.queue.len() as i32
    }
}

#[test]
fn test() {
    let mut obj = RecentCounter::new();
    assert_eq!(obj.ping(1), 1);
    assert_eq!(obj.ping(100), 2);
    assert_eq!(obj.ping(3001), 3);
    assert_eq!(obj.ping(3002), 3);
}