Laskettelukeskus

use std::io::stdin;
use std::cmp::max;

#[derive(Clone, Debug)]
struct Slope {
    children: Vec<usize>,
    plough_demand: u32
}
impl Slope {
    pub const DUMMY: Self = Self {
        children: Vec::new(),
        plough_demand: 0
    };
}

//possible stack overflow
fn get_total_plough_demand(parent_index: usize, all_slopes: &Vec<Slope>) -> u32 {
    let demand_of_children: u32 = all_slopes[parent_index].children
        .iter()
        .map(|&i| get_total_plough_demand(i, all_slopes))
        .sum();
    let demand_of_parent = all_slopes[parent_index].plough_demand;

    max(demand_of_children, demand_of_parent)
}

fn main() {
    let mut input = stdin().lines();

    let line = input.next().unwrap().unwrap();
    let path_count: usize = line.parse().unwrap();
    let mut slopes = vec![Slope::DUMMY; path_count];

    for line in (&mut input).take(path_count) {
        let indeces: Vec<usize> = line
            .unwrap()
            .split_whitespace()
            .map(|s| s.parse().unwrap())
            .map(|i: usize| i - 1)
            .collect();

        slopes[indeces[0]].children.push(indeces[1]);
    }

    let line = input.next().unwrap().unwrap();
    let plough_demands = line
        .split_whitespace()
        .map(|s| s.parse().unwrap());
    for (i, plough_demand) in plough_demands.enumerate() {
        slopes[i].plough_demand = plough_demand;
    }

    println!("{}", get_total_plough_demand(0, &slopes));
}

Test 1

Group: 1, 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 5 but the index is 184467440737...

Test 2

Group: 1, 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'called `Option::unwrap()` on a `None` value', input/code.rs:45:...

Test 3

Group: 1, 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'called `Option::unwrap()` on a `None` value', input/code.rs:45:...

Test 4

Group: 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 100000 but the index is 9810185...

Test 5

Group: 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 100000 but the index is 4090552...

Test 6

Group: 1, 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 100 but the index is 1844674407...

Test 7

Group: 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 100000 but the index is 1844674...

Test 8

Group: 1, 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'called `Option::unwrap()` on a `None` value', input/code.rs:45:...

Test 9

Group: 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 100000 but the index is 9999999...

Test 10

Group: 1, 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'called `Option::unwrap()` on a `None` value', input/code.rs:45:...

Test 11

Group: 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 100000 but the index is 1913557...

Test 12

Group: 1, 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'called `Option::unwrap()` on a `None` value', input/code.rs:45:...

Test 13

Group: 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 100000 but the index is 9999999...