Laskettelukeskus

warning: unused import: `min`
 --> input/code.rs:2:16
  |
2 |     cmp::{max, min},
  |                ^^^
  |
  = note: `#[warn(unused_imports)]` on by default

warning: value assigned to `needed_auras` is never read
  --> input/code.rs:16:13
   |
16 |     let mut needed_auras: i32 = needs[0];
   |             ^^^^^^^^^^^^
   |
   = help: maybe it is overwritten before being read?
   = note: `#[warn(unused_assignments)]` on by default

warning: 2 warnings emitted

use std::{
    cmp::{max, min},
    collections::HashMap,
    io,
};

fn main() {
    let n = input()[0];
    let mut abdict: HashMap<i32, Vec<i32>> = HashMap::new();
    for _ in 0..n - 1 {
        let ab = input();
        abdict.entry(ab[0]).or_insert(Vec::new()).push(ab[1]);
    }
    let needs = input();
    let mut newneeds = needs.clone();
    let mut needed_auras: i32 = needs[0];
    for rinne in (0..n).rev() {
        if abdict.contains_key(&rinne) {
            let children = abdict.get(&rinne).unwrap();
            let mut newneed = 0;
            for child in children {
                newneed += needs[(child - 1) as usize];
            }
            newneeds[rinne as usize] = max(newneed, needs[rinne as usize]);
        }
    }
    needed_auras = *newneeds.iter().max().unwrap();
    print!("{:?}", needed_auras);
}

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

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 1, 2

Verdict: WRONG ANSWER

Test 3

Group: 1, 2

Verdict: WRONG ANSWER

Test 4

Group: 2

Verdict: WRONG ANSWER

Test 5

Group: 2

Verdict: WRONG ANSWER

Test 6

Group: 1, 2

Verdict: ACCEPTED

Test 7

Group: 2

Verdict: ACCEPTED

Test 8

Group: 1, 2

Verdict: ACCEPTED

Test 9

Group: 2

Verdict: ACCEPTED

Test 10

Group: 1, 2

Verdict: WRONG ANSWER

Test 11

Group: 2

Verdict: WRONG ANSWER

Test 12

Group: 1, 2

Verdict: WRONG ANSWER

Test 13

Group: 2

Verdict: WRONG ANSWER