Maalaus

use std::io;

#[derive(PartialEq)]
enum MoveType {
    COLUMN,
    ROW
}

struct Move {
    index: usize,
    color: usize,
    move_type: MoveType,
}

fn main() {
    let mut input = String::new();
    io::stdin().read_line(&mut input).expect("failed to readline");
    let mut iter = input.trim().split_whitespace();
    let (n, m, k, q): (i64, i64, i64, i64) = (
        iter.next().unwrap().parse().unwrap(),
        iter.next().unwrap().parse().unwrap(),
        iter.next().unwrap().parse().unwrap(),
        iter.next().unwrap().parse().unwrap(),
    );

    let mut moves: Vec<Move> = Vec::new();

    for _ in 0..q {
        let mut input2 = String::new();
        io::stdin().read_line(&mut input2).expect("failed to readline");

        let inputs: Vec<&str> = input2.trim().split_whitespace().collect();
        let move_type: String = inputs[0].to_string();
        let index: usize = inputs[1].parse().unwrap();
        let color: usize = inputs[2].parse().unwrap();
        moves.push(Move {
            index: index - 1,
            color: color - 1,
            move_type: match move_type.as_str() {
                "R" => MoveType::ROW,
                "C" => MoveType::COLUMN,
                _ => panic!(),
            },
        });
    }

    // Movepool split into two different categories.
    let mut r_moves: Vec<usize> = Vec::new();
    let mut c_moves: Vec<usize> = Vec::new();
    let mut colors = vec![0; k as usize];
    let mut collective_painted: i64 = 0;
    let paint_cap: i64 = n*m;
    let mut c_size: i64 = 0;
    let mut r_size: i64 = 0;
    for pmove in moves.iter().rev() {
        if pmove.move_type == MoveType::ROW {
            if r_moves.contains(&pmove.index) {
                continue;
            }
            collective_painted += m - c_size as i64;
            colors[pmove.color] += m - c_size as i64;
            r_moves.push(pmove.index);
            r_size += 1;
        } else {
            if c_moves.contains(&pmove.index) {
                continue;
            }
            collective_painted += n - r_size as i64;
            colors[pmove.color] += n - r_size as i64;
            c_moves.push(pmove.index);
            c_size += 1;
        }
        if collective_painted == paint_cap {
            break;
        }
    }
    let mut output: String = String::new();
    for color in colors {
        output.push_str(color.to_string().as_str());
        output.push_str(" ");
    }
    output.pop();
    println!("{}", output);
}

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: TIME LIMIT EXCEEDED

Test 5

Group: 2, 4

Verdict: TIME LIMIT EXCEEDED

Test 6

Group: 3, 4

Verdict: ACCEPTED

Test 7

Group: 3, 4

Verdict: ACCEPTED

Test 8

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 9

Group: 4

Verdict: TIME LIMIT EXCEEDED

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: 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