Submission #73689472

---

Source Code Expand

use proconio::input;

fn main() {
    input! {
        s: String,
        t: String,
    }

    let s_chars: Vec<char> = s.chars().collect();
    let t_chars: Vec<char> = t.chars().collect();

    let s_no_a: Vec<char> = s_chars.iter().copied().filter(|&c| c != 'A').collect();
    let t_no_a: Vec<char> = t_chars.iter().copied().filter(|&c| c != 'A').collect();

    if s_no_a != t_no_a {
        println!("-1");
        return;
    }

    let get_a_segments = |chars: &[char]| -> Vec<usize> {
        let mut counts = Vec::new();
        let mut current_count = 0;
        for &c in chars {
            if c == 'A' {
                current_count += 1;
            } else {
                counts.push(current_count);
                current_count = 0;
            }
        }
        counts.push(current_count);
        counts
    };

    let s_a_counts = get_a_segments(&s_chars);
    let t_a_counts = get_a_segments(&t_chars);

    // 各セグメントの 'A' の数の差の絶対値を合計
    let ans: i64 = s_a_counts
        .iter()
        .zip(t_a_counts.iter())
        .map(|(&s, &t)| (s as i64 - t as i64).abs())
        .sum();

    println!("{}", ans);
}

Submission Info

Judge Result