Maalaus

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

fn read_next<'a, I>(iter: &mut I) -> u32
where
    I: Iterator<Item = &'a str>,
{
    iter.next()
        .expect("invalid input")
        .parse::<u32>()
        .expect("invalid integer")
}

fn main() {
    // Input
    let mut nmkq = String::new();
    io::stdin()
        .read_line(&mut nmkq)
        .expect("failed to read line");
    let mut nmk = nmkq.split_ascii_whitespace();
    let n = read_next(&mut nmk);
    let m = read_next(&mut nmk);
    let k = read_next(&mut nmk);
    let q = read_next(&mut nmk);

    let mut row_ops: HashMap<u32, (u32, u32)> = HashMap::new(); // row: time, color
    let mut col_ops: HashMap<u32, (u32, u32)> = HashMap::new(); // col: time, color

    for t in 0..q {
        let mut oic = String::new();
        io::stdin()
            .read_line(&mut oic)
            .expect("failed to read line");
        let mut oic = oic.split_ascii_whitespace();
        let o = oic.next().expect("too few arguments");
        let i = read_next(&mut oic);
        let c = read_next(&mut oic);
        match o {
            "R" => row_ops.insert(i - 1, (t, c - 1)),
            "C" => col_ops.insert(i - 1, (t, c - 1)),
            _ => panic!("illegal operation '{o}'"),
        };
    }

    let mut sums: Vec<u32> = vec![0; k as usize];
    // Rows
    for (r_time, r_color) in row_ops.values() {
        println!("{r_color} += {m}");
        sums[*r_color as usize] += m;
        col_ops
            .values()
            .filter(|(x_time, x_color)| x_time > r_time && x_color != r_color)
            .for_each(|_| {
                println!("{r_color} -= 1");
                sums[*r_color as usize] -= 1;
            });
    }
    // Cols
    for (c_time, c_color) in col_ops.values() {
        println!("{c_color} += {n}");
        sums[*c_color as usize] += n;
        row_ops
            .values()
            .filter(|(x_time, x_color)| x_time > c_time && x_color != c_color)
            .for_each(|_| {
                println!("{c_color} -= 1");
                sums[*c_color as usize] -= 1;
            });
    }

    for sum in sums {
        print!("{sum} ");
    }
    println!();
}

Test 1 (public)

Group: 1, 3, 4

Verdict: WRONG ANSWER

Test 2 (public)

Group: 3, 4

Verdict: WRONG ANSWER

Test 3

Group: 1, 3, 4

Verdict: WRONG ANSWER

Test 4

Group: 2, 4

Verdict: TIME LIMIT EXCEEDED

Test 5

Group: 2, 4

Verdict: TIME LIMIT EXCEEDED

Test 6

Group: 3, 4

Verdict: TIME LIMIT EXCEEDED

Test 7

Group: 3, 4

Verdict: WRONG ANSWER

Test 8

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 9

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 10

Group: 1, 3, 4

Verdict: WRONG ANSWER

Test 11

Group: 1, 3, 4

Verdict: WRONG ANSWER

Test 12

Group: 2, 4

Verdict: WRONG ANSWER

Test 13

Group: 2, 4

Verdict: WRONG ANSWER

Test 14

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 15

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 16

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 17

Group: 4

Verdict: TIME LIMIT EXCEEDED