Module a

Expand description

ยงABC 381 A - 11/22 String

refs: https://atcoder.jp/contests/abc381/tasks/abc381_a

use proconio::{input, marker::Chars};

fn main() {
    input! {
      n: usize,
      s: Chars,
    }

    let rl = run_length(&s);

    if s == ['/'] || rl == [('1', n / 2), ('/', 1), ('2', n / 2)] {
        println!("Yes");
    } else {
        println!("No");
    }
}

fn run_length(s: &[char]) -> Vec<(char, usize)> {
    s.iter().fold(Vec::new(), |mut acc, si| {
        if let Some((last_char, count)) = acc.last_mut() {
            if last_char == si {
                *count += 1;
            } else {
                acc.push((*si, 1));
            }
        } else {
            acc.push((*si, 1));
        }
        acc
    })
}