Maalaus

use std::io;
 
fn main() {
    let mut input = String::new();
    io::stdin().read_line(&mut input).expect("Failed to read input");
 
    let data: Vec<usize> = parse_data(&input);
 
    let mut rowvector: Vec<Vec<usize>> = vec!();
    let mut colvector: Vec<Vec<usize>> = vec!();
    let mut vec = Vec::new();
    vec.push(0);
 
    for _i in 0..data[3] {
        input.clear();
        io::stdin().read_line(&mut input).expect("Failed to read input");
 
        let colordata = parse_input(&input);
 
        if colordata.0 {
            let mut found_index = false;
            for row in &mut rowvector {
                if row[0] == colordata.1 {
                    found_index = true;
 
                    row.clear();
                    row.push(colordata.1);
                    row.push(colordata.2);
                    break;
                }
            }
 
            if !found_index {
                let mut vec = Vec::new();
                vec.push(colordata.1);
                vec.push(colordata.2);
                rowvector.push(vec);
            }
 
            for colnum in 0..colvector.len() {
                if !colvector[colnum][2..].contains(&colordata.1) {
                    colvector[colnum].push(colordata.1);
                    if colvector[colnum].len() == data[0]+2 {
                        colvector.remove(colnum);
                    }
                }
            }
        } else {
            let mut found_index = false;
            for col in &mut colvector {
                if col[0] == colordata.1 {
                    found_index = true;
 
                    col.clear();
                    col.push(colordata.1);
                    col.push(colordata.2);
                    break;
                }
            }
 
            if !found_index {
                let mut vec = Vec::new();
                vec.push(colordata.1);
                vec.push(colordata.2);
                colvector.push(vec);
            }
 
            for rownum in 0..rowvector.len() {
                if !rowvector[rownum][2..].contains(&colordata.1) {
                    rowvector[rownum].push(colordata.1);
                    if rowvector[rownum].len() == data[1]+2 {
                        rowvector.remove(rownum);
                    }
                }
            }
        }
    }
 
    let mut colorvector: Vec<usize> = vec!();
    for _i in 0..data[2] {
        colorvector.push(0);
    }
 
    for row in &rowvector {
        let color = row[1]-1;
        let missing = row.len()-2;
 
        colorvector[color] += data[1]-missing;
    }
 
    for col in &colvector {
        let color = col[1]-1;
        let missing = col.len()-2;
 
        colorvector[color] += data[0]-missing;
    }
 
    for color in colorvector {
        println!("{} ", color);
    }
}
 
pub fn parse_data(input: &String) -> Vec<usize> {
    let datavector = input.split_ascii_whitespace().map(|e| e.parse::<usize>().unwrap()).collect();
    return datavector;
}
 
pub fn parse_input(input: &String) -> (bool, usize, usize) {
    let datavector: Vec<&str> = input.split_ascii_whitespace().collect();
    let command = match datavector[0] {
        "R" => true,
        "C" => false,
        _ => false,
    };
 
    let index = datavector[1].parse::<usize>().unwrap();
    let color = datavector[2].parse::<usize>().unwrap();
 
    return (command, index-1, color);
}

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

thread 'main' panicked at input/code.rs:69:30:
index out of bounds: the len is 3 but the index is 3
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 12

Group: 2, 4

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:41:30:
index out of bounds: the len is 1 but the index is 2
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 13

Group: 2, 4

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:69:30:
index out of bounds: the len is 1 but the index is 1
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

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