Maalaus

use std::io;

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: Vec<(Option<u32>, u32)> = vec![(None, 0); n as usize]; // row: time, color
    let mut col_ops: Vec<(Option<u32>, u32)> = vec![(None, 0); m as usize]; // 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) as usize;
        let c = read_next(&mut oic);
        match o {
            "R" => row_ops[i - 1] = (Some(t), c - 1),
            "C" => col_ops[i - 1] = (Some(t), c - 1),
            _ => panic!("illegal operation '{o}'"),
        }
    }

    let mut sums: Vec<u32> = vec![0; k as usize];
    for y in 0..n {
        let l_row_op = row_ops[y as usize];
        for x in 0..m {
            let l_col_op = col_ops[x as usize];

            if l_row_op.0.is_some() && l_col_op.0.is_some() {
                if l_col_op.0.unwrap() > l_row_op.0.unwrap() {
                    sums[l_col_op.1 as usize] += 1;
                } else if l_col_op.0.unwrap() < l_row_op.0.unwrap() {
                    sums[l_row_op.1 as usize] += 1;
                }
            } else if l_col_op.0.is_some() {
                sums[l_col_op.1 as usize] += 1;
            } else if l_row_op.0.is_some() {
                sums[l_row_op.1 as usize] += 1;
            }
        }
    }
    for sum in sums {
        print!("{sum} ");
    }
    println!();
}

Test 1 (public)

Group: 1, 3, 4

Verdict: ACCEPTED

Test 2 (public)

Group: 3, 4

Verdict: TIME LIMIT EXCEEDED

Test 3

Group: 1, 3, 4

Verdict: ACCEPTED

Test 4

Group: 2, 4

Verdict: RUNTIME ERROR

Test 5

Group: 2, 4

Verdict: RUNTIME ERROR

Test 6

Group: 3, 4

Verdict: TIME LIMIT EXCEEDED

Test 7

Group: 3, 4

Verdict: TIME LIMIT EXCEEDED

Test 8

Group: 4

Verdict: RUNTIME ERROR

Test 9

Group: 4

Verdict: RUNTIME ERROR

Test 10

Group: 1, 3, 4

Verdict: ACCEPTED

Test 11

Group: 1, 3, 4

Verdict: ACCEPTED

Test 12

Group: 2, 4

Verdict: TIME LIMIT EXCEEDED

Test 13

Group: 2, 4

Verdict: TIME LIMIT EXCEEDED

Test 14

Group: 4

Verdict: RUNTIME ERROR

Test 15

Group: 4

Verdict: RUNTIME ERROR

Test 16

Group: 4

Verdict: RUNTIME ERROR

Test 17

Group: 4

Verdict: RUNTIME ERROR