leetcode-rust 0.1.2

Some solutions about leetcode in rust lang
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

#![allow(dead_code)]

pub fn find_restaurant(list1: Vec<String>, list2: Vec<String>) -> Vec<String> {
    use std::collections::HashMap;

    let mut map = HashMap::new();
    let mut min = 2000;
    let mut res = vec![];
    for (i, x) in list1.into_iter().enumerate() {
        map.insert(x, i);
    }
    for (i, x) in list2.into_iter().enumerate() {
        match map.get_mut(&x) {
            Some(v) => {
                if *v + i < min {
                    min = *v + i;
                    res.clear();
                    res.push(x);
                } else if *v + i == min {
                    res.push(x);
                }
            }
            None => {}
        }
    }
    res
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test1() {
        let list1 = vec!["Shogun", "Tapioca Express", "Burger King", "KFC"];
        let list1: Vec<String> = list1.into_iter().map(|x| x.to_string()).collect();
        let list2 = vec![
            "Piatti",
            "The Grill at Torrey Pines",
            "Hungry Hunter Steakhouse",
            "Shogun",
        ];
        let list2: Vec<String> = list2.into_iter().map(|x| x.to_string()).collect();
        assert_eq!(find_restaurant(list1, list2), vec!["Shogun"]);

        let list1 = vec!["Shogun", "Tapioca Express", "Burger King", "KFC"];
        let list1: Vec<String> = list1.into_iter().map(|x| x.to_string()).collect();
        let list2 = vec!["KFC", "Shogun", "Burger King"];
        let list2: Vec<String> = list2.into_iter().map(|x| x.to_string()).collect();
        assert_eq!(find_restaurant(list1, list2), vec!["Shogun"]);
    }
}