Rust Gym Leetcode 557 Reverse Words in a String III Rust Solution

557. Reverse Words in a String III

Given a string, you need to reverse the order of characters in each word within a sentence while still preserving whitespace and initial word order.

Example 1:

Input: "Let's take LeetCode contest"
Output: "s'teL ekat edoCteeL tsetnoc"

Note: In the string, each word is separated by single space and there will not be any extra space in the string.

557. Reverse Words in a String III

struct Solution;

impl Solution {
    fn reverse_words(s: String) -> String {
        let words: Vec<String> = s
            .split_whitespace()
            .map(|s| s.chars().rev().collect())
            .collect();
        let res: String = words.join(" ");
        res
    }
}

#[test]
fn test() {
    let s = "Let's take LeetCode contest".to_string();
    let r = "s'teL ekat edoCteeL tsetnoc".to_string();
    assert_eq!(Solution::reverse_words(s), r);
}