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

struct Solution;

use std::collections::HashMap;

impl Solution {
    fn subdomain_visits(cpdomains: Vec<String>) -> Vec<String> {
        let mut res: Vec<String> = vec![];
        let mut hm: HashMap<String, usize> = HashMap::new();
        for s in cpdomains {
            let (domains, count) = Self::parse(s);
            for domain in domains {
                let e = hm.entry(domain).or_default();
                *e += count;
            }
        }
        for (domain, count) in hm {
            res.push(format!("{} {}", count, domain));
        }
        res
    }

    fn parse(s: String) -> (Vec<String>, usize) {
        let mut domains: Vec<String> = vec![];
        let mut iter = s.split_whitespace();
        let count: usize = iter.next().unwrap().parse::<usize>().unwrap();
        let domain: String = iter.next().unwrap().parse::<String>().unwrap();
        for (i, c) in domain.chars().enumerate() {
            if c == '.' {
                let subdomain = &domain[i + 1..];
                domains.push(subdomain.to_string());
            }
        }
        domains.push(domain);
        (domains, count)
    }
}

#[test]
fn test() {
    let input: Vec<String> = vec_string!["9001 discuss.leetcode.com"];
    let mut answer: Vec<String> =
        vec_string!["9001 discuss.leetcode.com", "9001 leetcode.com", "9001 com"];
    let mut output = Solution::subdomain_visits(input);
    answer.sort();
    output.sort();
    assert_eq!(answer, output);

    let input: Vec<String> = vec_string![
        "900 google.mail.com",
        "50 yahoo.com",
        "1 intel.mail.com",
        "5 wiki.org"
    ];
    let mut answer: Vec<String> = vec_string![
        "901 mail.com",
        "50 yahoo.com",
        "900 google.mail.com",
        "5 wiki.org",
        "5 org",
        "1 intel.mail.com",
        "951 com"
    ];
    let mut output = Solution::subdomain_visits(input);
    answer.sort();
    output.sort();
    assert_eq!(answer, output);
}