lib.rs

// https://leetcode.com/problems/count-and-say/

pub struct Solution;

struct CountedDigit {
    count: i32,
    digit: char,
}

impl Solution {
    pub fn count_and_say(n: i32) -> String {
        let mut result = 1.to_string();

        for _ in 2..=n {
            let pairs = Solution::to_counted_digits(&result);
            result = Solution::to_number(&pairs);
        }

        result
    }

    /// A helper function that maps a str slice of digits
    /// to pairs of the digits and their frequencies.
    ///
    /// For example, if you call this function with "22444",
    /// then it maps it to
    /// vec![CountedDigit {count: 2, digit: '2'}, CountedDigit {count: 3, digit: '4'}].
    ///
    /// # Panics
    /// Panics when digits.len() < 1
    fn to_counted_digits(digits: &str) -> Vec<CountedDigit> {
        let mut res = Vec::new();

        let mut prev_digit = digits.chars().next().unwrap();
        let mut count = 0;

        for current_digit in digits.chars() {
            if current_digit != prev_digit {
                res.push(CountedDigit {
                    count,
                    digit: prev_digit,
                });
                count = 0;
            }
            prev_digit = current_digit;
            count += 1;
        }
        res.push(CountedDigit {
            count,
            digit: prev_digit,
        });

        res
    }

    fn to_number(pairs: &[CountedDigit]) -> String {
        pairs.iter().fold(
            String::with_capacity(pairs.len() * 2),
            |acc, CountedDigit { count, digit }| format!("{}{}{}", acc, count, digit),
        )
    }
}

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

    #[test]
    fn n1() {
        assert_eq!(Solution::count_and_say(1), String::from("1"));
    }

    #[test]
    fn n4() {
        assert_eq!(Solution::count_and_say(4), String::from("1211"));
    }

    #[test]
    fn n30() {
        assert_eq!(Solution::count_and_say(30), String::from("3113112221131112311332111213122112311311123112111331121113122112132113121113222112311311221112131221123113112221121113311211131122211211131221131211132221121321132132212321121113121112133221123113112221131112212211131221121321131211132221123113112221131112311332211211133112111311222112111312211311123113322112111312211312111322212321121113121112133221121321132132211331121321132213211231132132211211131221232112111312212221121123222112311311222113111231133211121321321122111312211312111322211213211321322123211211131211121332211231131122211311123113321112131221123113111231121123222112111331121113112221121113122113111231133221121113122113121113221112131221123113111231121123222112111312211312111322212321121113121112131112132112311321322112111312212321121113122122211211232221121321132132211331121321231231121113112221121321132132211322132113213221123113112221133112132123222112111312211312112213211231132132211211131221131211322113321132211221121332211231131122211311123113321112131221123113111231121113311211131221121321131211132221123113112211121312211231131122211211133112111311222112111312211312111322211213211321223112111311222112132113213221133122211311221122111312211312111322212321121113121112131112132112311321322112111312212321121113122122211211232221121321132132211331121321231231121113112221121321132132211322132113213221123113112221133112132123222112111312211312112213211231132132211211131221131211322113321132211221121332211213211321322113311213212312311211131122211213211331121321123123211231131122211211131221131112311332211213211321223112111311222112132113213221123123211231132132211231131122211311123113322112111312211312111322111213122112311311123112112322211213211321322113312211223113112221121113122113111231133221121321132132211331222113321112131122211332113221122112133221123113112221131112311332111213122112311311123112111331121113122112132113121113222112311311221112131221123113112221121113311211131122211211131221131211132221121321132132212321121113121112133221123113112221131112311332111213122112311311123112112322211322311311222113111231133211121312211231131112311211232221121113122113121113222123211211131221132211131221121321131211132221123113112211121312211231131122113221122112133221121321132132211331121321231231121113121113122122311311222113111231133221121113122113121113221112131221123113111231121123222112132113213221133112132123123112111312211322311211133112111312211213211311123113223112111321322123122113222122211211232221121113122113121113222123211211131211121311121321123113213221121113122123211211131221121311121312211213211321322112311311222113311213212322211211131221131211221321123113213221121113122113121113222112131112131221121321131211132221121321132132211331121321232221123113112221131112311322311211131122211213211331121321122112133221121113122113121113222123112221221321132132211231131122211331121321232221121113122113121113222123211211131211121332211213111213122112132113121113222112132113213221232112111312111213322112132113213221133112132123123112111311222112132113311213211221121332211231131122211311123113321112131221123113112221132231131122211211131221131112311332211213211321223112111311222112132113212221132221222112112322211211131221131211132221232112111312111213111213211231131112311311221122132113213221133112132123222112311311222113111231132231121113112221121321133112132112211213322112111312211312111322212321121113121112131112132112311321322112111312212321121113122122211211232221121311121312211213211312111322211213211321322123211211131211121332211213211321322113311213211322132112311321322112111312212321121113122122211211232221121321132132211331121321231231121113112221121321133112132112312321123113112221121113122113111231133221121321132122311211131122211213211321222113222122211211232221123113112221131112311332111213122112311311123112111331121113122112132113121113222112311311221112131221123113112221121113311211131122211211131221131211132221121321132132212321121113121112133221123113112221131112311332111213213211221113122113121113222112132113213221232112111312111213322112132113213221133112132123123112111312211322311211133112111312212221121123222112132113213221133112132123222113223113112221131112311332111213122112311311123112112322211211131221131211132221232112111312111213111213211231132132211211131221131211221321123113213221123113112221131112211322212322211231131122211322111312211312111322211213211321322113311213211331121113122122211211132213211231131122212322211331222113112211"));
    }
}