Card game

#![allow(unused)]

use core::num;
use std::{io, process::exit};

fn take_int() -> usize {
    let mut input = String::new();
    io::stdin().read_line(&mut input).unwrap();
    return input.trim().parse::<usize>().unwrap();
}

fn take_vector() -> Vec<usize> {
    let mut input = String::new();
    io::stdin().read_line(&mut input).unwrap();
    let arr: Vec<usize> = input
        .trim()
        .split_whitespace()
        .map(|x| x.parse::<usize>().unwrap())
        .collect();
    return arr;
}

fn main() {
    let mut num_cards: usize = take_int();
    let mut cards: Vec<usize> = take_vector();

    if num_cards < 3 {
        println!("0");
        exit(0);
    }

    let mut points: Vec<usize> = Vec::with_capacity(num_cards);
    points.push(0);
    points.push(0);
    points.push(cards[1]);
    let mut max: usize = cards[1];

    for choice in 3..num_cards-1 {
        let taking = cards[choice-1] + points[choice - 3];
        let not_taking = points[choice - 1];
        if taking > not_taking {
            points.push(taking);
            max = taking;
        } else {
            points.push(not_taking);
        }
    }
    println!("{}", max);
}

Test 1

Verdict: WRONG ANSWER

Test 2

Verdict: WRONG ANSWER

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: WRONG ANSWER

Test 6

Verdict: ACCEPTED

Test 7

Verdict: ACCEPTED

Test 8

Verdict: WRONG ANSWER

Test 9

Verdict: ACCEPTED

Test 10

Verdict: ACCEPTED

Test 11

Verdict: ACCEPTED

Test 12

Verdict: ACCEPTED

Test 13

Verdict: ACCEPTED

Test 14

Verdict: ACCEPTED

Test 15

Verdict: ACCEPTED