Maalaus

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

fn main() {
    let mut input: String = String::new();
    io::stdin().read_line(&mut input).unwrap();
    let input: Vec<usize> = input.trim().split(" ").map(|s| s.parse::<usize>().unwrap()).collect();

    let operations: Vec<(String, usize)> = parse_operations(&input);
    count_colors(operations, input);
}

fn parse_operations(input: &Vec<usize>) -> Vec<(String, usize)> {
    let mut operations: Vec<(String, usize)> = Vec::new();

    for _ in 0..input[3] {
        let mut operation: String = String::new();
        io::stdin().read_line(&mut operation).unwrap();
        let operation: Vec<&str> = operation.trim().split(" ").collect();
        let operation: (String, usize) = (format!("{}{}", operation[0], operation[1]), operation[2].parse::<usize>().unwrap());
        
        operations.retain(|v: &(String, usize)| v.0 != operation.0);
        operations.push(operation);
    }

    return operations;
}

fn count_colors(operations: Vec<(String, usize)>, input: Vec<usize>) {
    let mut colors: HashMap<usize, usize> = HashMap::new();
    let mut rows: usize = 0;
    let mut columns: usize = 0;

    let operation_count: usize = operations.len();
    for i in 0..operation_count {
        let operation: &(String, usize) = &operations[operation_count - i - 1];

        let amount: usize;
        if operation.0.starts_with("R") {
            amount = input[1] - columns;
            rows += 1;
        } else {
            amount = input[0] - rows;
            columns += 1;
        }

        match colors.get_mut(&operation.1) {
            Some(color) => { *color += amount; }
            None => { colors.insert(operation.1, amount); }
        }
    }
    print_result(colors, input[2]);
}

fn print_result(colors: HashMap<usize, usize>, k: usize) {
    for i in 0..k {
        match colors.get(&(i+1)) {
            Some(color) => { print!("{color} "); }
            None => { print!("0 "); }
        }
    }
}

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

Test 13

Group: 2, 4

Verdict: TIME LIMIT EXCEEDED

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