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

struct Solution;
use std::usize;

impl Solution {
    fn suggested_products(mut products: Vec<String>, search_word: String) -> Vec<Vec<String>> {
        let mut res = vec![];
        products.sort_unstable();
        let n = products.len();
        let mut prefix = "".to_string();
        let mut start = 0;
        for c in search_word.chars() {
            prefix.push(c);
            start = start
                + products[start..]
                    .binary_search(&prefix)
                    .unwrap_or_else(|p| p);
            let mut list: Vec<String> = vec![];
            let end = usize::min(start + 3, n);
            for i in start..end {
                if products[i].starts_with(&prefix) {
                    list.push(products[i].clone());
                }
            }
            res.push(list);
        }
        res
    }
}

#[test]
fn test() {
    let products = vec_string!["mobile", "mouse", "moneypot", "monitor", "mousepad"];
    let search_word = "mouse".to_string();
    let res: Vec<Vec<String>> = vec_vec_string![
        ["mobile", "moneypot", "monitor"],
        ["mobile", "moneypot", "monitor"],
        ["mouse", "mousepad"],
        ["mouse", "mousepad"],
        ["mouse", "mousepad"]
    ];
    assert_eq!(Solution::suggested_products(products, search_word), res);
    let products = vec_string!["havana"];
    let search_word = "havana".to_string();
    let res: Vec<Vec<String>> = vec_vec_string![
        ["havana"],
        ["havana"],
        ["havana"],
        ["havana"],
        ["havana"],
        ["havana"]
    ];
    assert_eq!(Solution::suggested_products(products, search_word), res);
    let products = vec_string!["bags", "baggage", "banner", "box", "cloths"];
    let search_word = "bags".to_string();
    let res: Vec<Vec<String>> = vec_vec_string![
        ["baggage", "bags", "banner"],
        ["baggage", "bags", "banner"],
        ["baggage", "bags"],
        ["bags"]
    ];
    assert_eq!(Solution::suggested_products(products, search_word), res);
    let products = vec_string!["havana"];
    let search_word = "tatiana".to_string();
    let res: Vec<Vec<String>> = vec_vec_string![[], [], [], [], [], [], []];
    assert_eq!(Solution::suggested_products(products, search_word), res);
}