Maalaus

warning: unused import: `std::collections::HashMap`
 --> input/code.rs:2:5
  |
2 | use std::collections::HashMap;
  |     ^^^^^^^^^^^^^^^^^^^^^^^^^
  |
  = note: `#[warn(unused_imports)]` on by default

warning: unused variable: `i`
  --> input/code.rs:32:9
   |
32 |     for i in 0..q {
   |         ^ help: if this is intentional, prefix it with an underscore: `_i`
   |
   = note: `#[warn(unused_variables)]` on by default

warning: field `influence` is never read
  --> input/code.rs:14:5
   |
10 | struct Move {
   |        ---- field in this struct
...
14 |     influence: i64,
   |     ^^^^^^^^^
   |
   = note: `#[warn(dead_code)]` on by default

warning: 3 warnings emitted

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

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

struct Move {
    index: usize,

    color: usize,
    influence: i64,

    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 i 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!(),
            },
            influence: match move_type.as_str() {
                "R" => m,
                "C" => n,
                _ => 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 = 0;
    let paint_cap = n*m;
    for pmove in moves.iter().rev() {
        if pmove.move_type == MoveType::ROW {
            if r_moves.contains(&pmove.index) {
                continue;
            }
            let c_size = c_moves.len();
            collective_painted += m - c_size as i64;
            colors[pmove.color] += m - c_size as i64;
            r_moves.push(pmove.index);
        } else {
            if c_moves.contains(&pmove.index) {
                continue;
            }
            let r_size = r_moves.len();
            collective_painted += n - r_size as i64;
            colors[pmove.color] += n - r_size as i64;
            c_moves.push(pmove.index);
        }
        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