Maalaus

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

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

warning: 2 warnings emitted

use std::io;
use std::collections::{HashSet, HashMap};
use rand::*;
 
fn main() {
    let mut input = String::new();
    io::stdin().read_line(&mut input).expect("Failed to read input");
 
    let startdata: Vec<usize> = input.split_ascii_whitespace().map(|e| e.parse::<usize>().unwrap()).collect();
 
    let mut instructions: Vec<(usize, usize, bool)> = vec!();
 
    // Kerää ohjeet vektoriin
    for instructionindex in 0..startdata[3] {
        let mut stepinput = String::new();
        io::stdin().read_line(&mut stepinput).expect("Failed to read input");
        let instruction = parse_input(&stepinput);
        instructions.push(instruction);

        /*let op = random_bool(0.5);
        let index = random_range(0..startdata[1]);
        let color = random_range(0..startdata[2]);
        let instruction = (index, color, op);
        instructions.push(instruction);*/
    }

    let mut instructionsvec = instructions.into_iter().rev().collect::<Vec<(usize, usize, bool)>>();
    dedup(&mut instructionsvec);
    let instructionlist = instructionsvec.into_iter().rev().collect::<Vec<(usize, usize, bool)>>();

    if startdata[2] == 1 {
        let mut rowsfilled: usize = 0;
        let mut colsfilled: usize = 0;
        let mut color: usize = 0;
        for instruction in instructionlist {
            if instruction.2 {
                rowsfilled += 1;
                color += startdata[1];
                color -= colsfilled;
            } else {
                colsfilled += 1;
                color += startdata[0];
                color -= rowsfilled;
            }
        }
        println!("{}", color);
    } else {
        let mut rowvector: Vec<(usize, usize)> = vec!();
        let mut colvector: Vec<(usize, usize)> = vec!();
        let mut colors: HashMap<usize, usize> = HashMap::new();

        for instruction in instructionlist {
            //println!("Running: {} {} {}", instruction.0, instruction.1, instruction.2);
            let mut found_index = false;
            if instruction.2 {
                for row in &mut rowvector {
                    if row.0 == instruction.1 {
                        found_index = true;
                        row.1 += 1;
                    }
                }
                if !found_index {
                    rowvector.push((instruction.1, 1));
                }

                colors.entry(instruction.1).and_modify(|val| *val += startdata[1]).or_insert(startdata[1]);
                for col in &colvector {
                    colors.entry(col.0).and_modify(|val| *val -= col.1);
                }
            } else {
                for col in &mut colvector {
                    if col.0 == instruction.1 {
                        found_index = true;
                        col.1 += 1;
                    }
                }
                if !found_index {
                    colvector.push((instruction.1, 1));
                }
                colors.entry(instruction.1).and_modify(|val| *val += startdata[1]).or_insert(startdata[1]);
                for row in &rowvector {
                    colors.entry(row.0).and_modify(|val| *val -= row.1);
                }
            }
        }
        for key in 0..startdata[2] {
            match colors.get(&key) {
                Some(s) => {println!("{}", s);}
                None => {println!("0");}
            }
        }
    }
}
 
pub fn parse_input(input: &String) -> (usize, usize, bool) {
    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 (index-1, color-1, command);
}
 
fn dedup(v: &mut Vec<(usize, usize, bool)>) {
    let mut set = HashSet::new();
    v.retain(|x| set.insert((x.0, x.2)));
}

Test 1 (public)

Group: 1, 3, 4

Verdict: WRONG ANSWER

Test 2 (public)

Group: 3, 4

Verdict: ACCEPTED

Test 3

Group: 1, 3, 4

Verdict: ACCEPTED

Test 4

Group: 2, 4

Verdict: ACCEPTED

Test 5

Group: 2, 4

Verdict: ACCEPTED

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

Test 11

Group: 1, 3, 4

Verdict: WRONG ANSWER

Test 12

Group: 2, 4

Verdict: ACCEPTED

Test 13

Group: 2, 4

Verdict: ACCEPTED

Test 14

Group: 4

Verdict: ACCEPTED

Test 15

Group: 4

Verdict: ACCEPTED

Test 16

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 17

Group: 4

Verdict: TIME LIMIT EXCEEDED