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 colorvector: Vec<usize> = vec!();

    for _i in 0..data[0] {
        let mut vec = Vec::new();
        vec.push(0);
        rowvector.push(vec);
    }

    for _i in 0..data[1] {
        let mut vec = Vec::new();
        vec.push(0);
        colvector.push(vec);
    }

    for _i in 0..data[2] {
        colorvector.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 {
            if rowvector[colordata.1][0] == colordata.2 {
                colorvector[colordata.1] += rowvector[colordata.1].len()-1;
                rowvector[colordata.1].clear();
                rowvector[colordata.1].push(colordata.2);
                continue;
            }

            let mut remove: Vec<usize> = vec!();

            for _i in 0..data[2]+1 {
                remove.push(0);
            }

            for column in &mut colvector {
                if column[0] != 0 {
                    if !column[1..].contains(&colordata.1) {
                        remove[column[0]] += 1;
                        if column[0] != colordata.1 {
                            column.push(colordata.1);
                        }
                    }
                }
            }

            colorvector[colordata.2-1] += data[1];

            for color in 0..data[2] {
                colorvector[color] -= remove[color+1];
            }
        } else {
            if colvector[colordata.1][0] == colordata.2 {
                colorvector[colordata.1] += colvector[colordata.1].len()-1;
                colvector[colordata.1].clear();
                colvector[colordata.1].push(colordata.2);
                continue;
            }

            let mut remove: Vec<usize> = vec!();

            for _i in 0..data[2]+1 {
                remove.push(0);
            }

            for row in &mut rowvector {
                if row[0] != 0 {
                    if !row[1..].contains(&colordata.1) {
                        remove[row[0]] += 1;
                        if row[0] != colordata.1 {
                            row.push(colordata.1);
                        }
                    }
                }
            }

            colorvector[colordata.2-1] += data[0];

            for color in 0..data[2] {
                colorvector[color] -= remove[color+1];
            }
        }
    }
}

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, color);
}

Test 1 (public)

Group: 1, 3, 4

Verdict: WRONG ANSWER

Test 2 (public)

Group: 3, 4

Verdict: RUNTIME ERROR

Test 3

Group: 1, 3, 4

Verdict: RUNTIME ERROR

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

Test 4

Group: 2, 4

Verdict: RUNTIME ERROR

Test 5

Group: 2, 4

Verdict: RUNTIME ERROR

Test 6

Group: 3, 4

Verdict: RUNTIME ERROR

Test 7

Group: 3, 4

Verdict: RUNTIME ERROR

Test 8

Group: 4

Verdict: RUNTIME ERROR

Test 9

Group: 4

Verdict: RUNTIME ERROR

Test 10

Group: 1, 3, 4

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:36:25:
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 11

Group: 1, 3, 4

Verdict: RUNTIME ERROR

thread 'main' panicked at input/code.rs:66:25:
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 12

Group: 2, 4

Verdict: RUNTIME ERROR

Test 13

Group: 2, 4

Verdict: RUNTIME ERROR

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