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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79

use std::collections::BTreeMap;
use std::collections::HashMap;

const SECONDS: [usize; 3] = [60, 3600, 86400];

#[derive(Default)]
struct TweetCounts {
    data: HashMap<String, BTreeMap<usize, usize>>,
}

impl TweetCounts {
    fn new() -> Self {
        TweetCounts::default()
    }

    fn record_tweet(&mut self, tweet_name: String, time: i32) {
        let time = time as usize;
        *self
            .data
            .entry(tweet_name)
            .or_default()
            .entry(time)
            .or_default() += 1;
    }

    fn get_tweet_counts_per_frequency(
        &self,
        freq: String,
        tweet_name: String,
        start_time: i32,
        end_time: i32,
    ) -> Vec<i32> {
        let seconds = SECONDS[Self::freq_to_index(&freq)];
        let start_time = start_time as usize;
        let end_time = end_time as usize;
        let n = (end_time - start_time) / seconds + 1;
        let mut res = vec![0; n];

        if let Some(counts) = self.data.get(&tweet_name) {
            for (&t, &c) in counts {
                let i = (t - start_time) / seconds;
                if t >= start_time && t <= end_time {
                    res[i] += c as i32;
                }
            }
        }
        res
    }

    fn freq_to_index(freq: &str) -> usize {
        match freq {
            "minute" => 0,
            "hour" => 1,
            "day" => 2,
            _ => panic!(),
        }
    }
}

#[test]
fn test() {
    let mut obj = TweetCounts::new();
    obj.record_tweet("tweet3".to_string(), 0);
    obj.record_tweet("tweet3".to_string(), 60);
    obj.record_tweet("tweet3".to_string(), 10);
    assert_eq!(
        obj.get_tweet_counts_per_frequency("minute".to_string(), "tweet3".to_string(), 0, 59),
        vec![2]
    );
    assert_eq!(
        obj.get_tweet_counts_per_frequency("minute".to_string(), "tweet3".to_string(), 0, 60),
        vec![2, 1]
    );
    obj.record_tweet("tweet3".to_string(), 120);
    assert_eq!(
        obj.get_tweet_counts_per_frequency("hour".to_string(), "tweet3".to_string(), 0, 210),
        vec![4]
    );
}