Kortit I

warning: variable does not need to be mutable
  --> input/code.rs:13:13
   |
13 |         let mut values: Vec<i32> = {
   |             ----^^^^^^
   |             |
   |             help: remove this `mut`
   |
   = note: `#[warn(unused_mut)]` on by default

warning: function `generateAnswer` should have a snake case name
  --> input/code.rs:31:4
   |
31 | fn generateAnswer(total_cards: i32, a_points: i32, b_points: i32) -> Option<[Vec<i32>; 2]> {
   |    ^^^^^^^^^^^^^^ help: convert the identifier to snake case: `generate_answer`
   |
   = note: `#[warn(non_snake_case)]` on by default

warning: 2 warnings emitted

use std::io;

fn main() {
    let stdin = io::stdin();

    let t: i32 = {
        let mut input: String = String::new();
        stdin.read_line(&mut input).unwrap();
        input.trim().parse().unwrap()
    };

    for _ in 0..t {
        let mut values: Vec<i32> = {
            let mut input: String = String::new();
            stdin.read_line(&mut input).unwrap();
            input.trim().split_whitespace().map(|x| x.parse::<i32>().unwrap()).collect()
        };

        let answer = generateAnswer(values[0], values[1], values[2]);
        if answer.is_none() {
            println!("NO");
        } else {
            println!("YES");
            for row in answer.unwrap() {
                println!("{}", row.iter().map(|x| x.to_string()).collect::<Vec<String>>().join(" "));
            }
        }
    }
}

fn generateAnswer(total_cards: i32, a_points: i32, b_points: i32) -> Option<[Vec<i32>; 2]> {
    let total_points = a_points + b_points;

    if total_points != 0 && (a_points <= 0 || b_points <= 0) {
        return None;
    }

    if total_points > total_cards {
        return None;
    }

    let mut a_plays = Vec::new();
    let mut b_plays = Vec::new();

    if total_points > 0 {
        for i in 1..=a_points {
            a_plays.push(i + 1);
            b_plays.push(i);
        }
        for i in (a_points + 1)..=(a_points + b_points - 1) { // minus 1 is because the last case will be handled separately
            a_plays.push(i);
            b_plays.push(i + 1);
        }

        // Lastly, A loses to B once.
        a_plays.push(1);
        b_plays.push(a_points + b_points);
    }

    // Play all remaining draws.
    for i in (a_points + b_points + 1)..=total_cards {
        a_plays.push(i);
        b_plays.push(i);
    }

    Some([a_plays, b_plays])
}

Test 1

Group: 1, 2, 3

Verdict: WRONG ANSWER

Test 2

Group: 2, 3

Verdict: WRONG ANSWER

Test 3

Group: 3

Verdict: WRONG ANSWER

Test 4

Group: 3

Verdict: WRONG ANSWER

Test 5

Group: 3

Verdict: WRONG ANSWER

Test 6

Group: 3

Verdict: WRONG ANSWER

Test 7

Group: 3

Verdict: WRONG ANSWER

Test 8

Group: 3

Verdict: WRONG ANSWER

Test 9

Group: 3

Verdict: WRONG ANSWER

Test 10

Group: 3

Verdict: WRONG ANSWER

Test 11

Group: 3

Verdict: WRONG ANSWER

Test 12

Group: 3

Verdict: WRONG ANSWER

Test 13

Group: 3

Verdict: WRONG ANSWER

Test 14

Group: 3

Verdict: WRONG ANSWER

Test 15

Group: 3

Verdict: WRONG ANSWER

Test 16

Group: 3

Verdict: WRONG ANSWER

Test 17

Group: 3

Verdict: WRONG ANSWER

Test 18

Group: 3

Verdict: WRONG ANSWER

Test 19

Group: 3

Verdict: WRONG ANSWER

Test 20

Group: 3

Verdict: WRONG ANSWER