Ruudukko

warning: unused variable: `i`
  --> input/code.rs:11:9
   |
11 |     for i in 0..koko {
   |         ^ help: consider prefixing with an underscore: `_i`
   |
   = note: #[warn(unused_variables)] on by default

warning: unused variable: `j`
  --> input/code.rs:13:13
   |
13 |         for j in 0..koko {
   |             ^ help: consider prefixing with an underscore: `_j`

warning: unused variable: `i`
  --> input/code.rs:20:9
   |
20 |     for i in 0..koko {
   |         ^ help: consider prefixing with an underscore: `_i`

use std::io;

fn main() {
    let mut koko_str = String::new();
    io::stdin()
        .read_line(&mut koko_str).unwrap();

    let koko = koko_str
        .trim().parse::<usize>().unwrap();
    let mut ruudukko = Vec::with_capacity(koko);
    for i in 0..koko {
        let mut sar = Vec::with_capacity(koko);
        for j in 0..koko {
            sar.push(0);
        }
        ruudukko.push(sar);
    }

    let mut sar_pienin_sallittu = Vec::with_capacity(koko);
    for i in 0..koko {
        sar_pienin_sallittu.push(1);
    }
    
    for i in 0..koko {
        let mut riv_pienin_sallittu = 1;
        for j in 0..koko {
            let valittu = std::cmp::max(riv_pienin_sallittu, sar_pienin_sallittu[j]);
            if valittu == sar_pienin_sallittu[j] {
                sar_pienin_sallittu[j] += 1;
            }
            if valittu == riv_pienin_sallittu {
                riv_pienin_sallittu += 1;
            }
            ruudukko[i][j] = valittu;
        }
    }

    for i in 0..koko {
        for j in 0..koko {
            print!("{}", ruudukko[i][j]);
            if j != koko - 1 {
                print!(" ");
            }
        }
        println!("");
    }
}

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: WRONG ANSWER

Test 4

Verdict: WRONG ANSWER

Test 5

Verdict: WRONG ANSWER

Test 6

Verdict: WRONG ANSWER