Maalaus

use std::collections::HashSet;
use std::io::{self, Read};

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 nmkq = nmkq.split_ascii_whitespace();
    let n = read_next(&mut nmkq);
    let m = read_next(&mut nmkq);
    let k = read_next(&mut nmkq);
    let _q = read_next(&mut nmkq);

    let mut row_ops: HashSet<u32> = HashSet::new();
    let mut col_ops: HashSet<u32> = HashSet::new();
    let mut sums: Vec<u64> = vec![0; k as usize];

    let mut input = String::new();
    io::stdin()
        .read_to_string(&mut input)
        .expect("failed to read input");
    let input = input.split("\n").collect::<Vec<_>>().into_iter().rev();

    for instr in input {
        if instr.len() == 0 {
            continue;
        }
        let mut oic = instr.split_ascii_whitespace();
        let o = oic.next().expect("too few arguments");
        let i = read_next(&mut oic) - 1;
        let c = read_next(&mut oic) - 1;
        match o {
            "R" => {
                if row_ops.get(&i).is_some() {
                    continue;
                }
                sums[c as usize] += m as u64 - col_ops.len() as u64;
                row_ops.insert(i);
            }
            "C" => {
                if col_ops.get(&i).is_some() {
                    continue;
                }
                sums[c as usize] += n as u64 - row_ops.len() as u64;
                col_ops.insert(i);
            }
            _ => panic!("invalid operation"),
        };
    }

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

Test 1 (public)

Group: 1, 3, 4

Verdict: ACCEPTED

Test 2 (public)

Group: 3, 4

Verdict: ACCEPTED

Test 3

Group: 1, 3, 4

Verdict: ACCEPTED

Test 4

Group: 2, 4

Verdict: ACCEPTED

Test 5

Group: 2, 4

Verdict: ACCEPTED

Test 6

Group: 3, 4

Verdict: ACCEPTED

Test 7

Group: 3, 4

Verdict: ACCEPTED

Test 8

Group: 4

Verdict: ACCEPTED

Test 9

Group: 4

Verdict: ACCEPTED

Test 10

Group: 1, 3, 4

Verdict: ACCEPTED

Test 11

Group: 1, 3, 4

Verdict: ACCEPTED

Test 12

Group: 2, 4

Verdict: ACCEPTED

Test 13

Group: 2, 4

Verdict: ACCEPTED

Test 14

Group: 4

Verdict: ACCEPTED

Test 15

Group: 4

Verdict: ACCEPTED

Test 16

Group: 4

Verdict: ACCEPTED

Test 17

Group: 4

Verdict: ACCEPTED