【PAT】 1002 写出这个数 Rust Solution

读入一个正整数 n，计算其各位数字之和，用汉语拼音写出和的每一位数字。

输入格式：

每个测试输入包含 1 个测试用例，即给出自然数 n 的值。这里保证 n 小于 10​100​​。

输出格式：

在一行内输出 n 的各位数字之和的每一位，拼音数字间有 1 空格，但一行中最后一个拼音数字后没有空格。

输入样例：

1234567890987654321123456789

 

输出样例：

yi san wu

use std::collections::HashMap;
use std::io;

pub fn  write_the_number(n: String) -> Vec<String> {

    let mut  temp_hashmap = HashMap::new();
    temp_hashmap.insert(0u8, "ling");
    temp_hashmap.insert(1,"yi");
    temp_hashmap.insert(2, "er");
    temp_hashmap.insert(3, "san");
    temp_hashmap.insert(4, "si");
    temp_hashmap.insert(5, "wu");
    temp_hashmap.insert(6, "liu");
    temp_hashmap.insert(7, "qi");
    temp_hashmap.insert(8, "ba");
    temp_hashmap.insert(9, "jiu");


    let sum = n
        .chars()
        .map(|temp| {
        temp.to_string()
            .as_str()
            .parse::<u32>()
            .unwrap()
        })
        .collect::<Vec<u32>>()
        .iter()
        .sum::<u32>();

    let ret = sum
        .to_string()
        .chars()
        .map(|i| {
            let index = i.to_string().parse::<u8>().unwrap();
            temp_hashmap.get(&index)
                .unwrap()
                .to_string()
        }).collect::<Vec<String>>();

    ret
}


fn main() {
    let mut input_string = String::new();

    io::stdin().read_line(&mut input_string)
        .expect("Failed to read line");

    let n = input_string.trim().to_string();
    let ret = write_the_number(n);
    for i in 0..ret.len() {
        if i == ret.len() - 1 {
            print!("{}", ret.get(i).unwrap());
        }else {
            print!("{} ", ret.get(i).unwrap());
        }
    }
}

GIthub Link: https://github.com/DaviRain-Su/my_rust_road/blob/master/pat_solution/src/lib.rs