Laskettelukeskus

use std::{collections::HashMap, io::*};
 
#[derive(Debug, Clone)]
struct Slope {
    connected: Vec<i32>,
    times_to_plow: i64,
}
 
impl Slope {
    fn calc_times_to_plow(&self, slopes: &HashMap<i32, Slope>) -> i64 {
        let mut sum = 0;
 
        for slope_id in &self.connected {
            let slope = slopes.get(&slope_id).unwrap();
            sum += slope.calc_times_to_plow(slopes);
        }
 
        sum.max(self.times_to_plow)
    }
}
 
fn main() -> Result<()> {
    let mut input = String::new();
    stdin().read_line(&mut input)?;
 
    let n: i32 = input.trim().parse().unwrap();
    let mut slopes = HashMap::new();
 
    for id in 0..n {
        slopes.insert(
            id + 1,
            Slope {
                connected: Vec::new(),
                times_to_plow: 0,
            },
        );
    }
    //println!("{:#?}", slopes);
 
    for _ in 1..n {
        let mut connection = String::new();
        stdin().read_line(&mut connection)?;
        let mut s = connection.split(" ");
        let id: i32 = s.next().unwrap().trim().parse().unwrap();
        let connection: i32 = s.next().unwrap().trim().parse().unwrap();
        slopes.get_mut(&id).unwrap().connected.push(connection);
    }
 
    let mut input = String::new();
    stdin().read_line(&mut input)?;
    input.split(" ").enumerate().for_each(|(id, s)| {
        slopes.get_mut(&(id as i32 + 1)).unwrap().times_to_plow = s.trim().parse().unwrap();
    });
 
    //println!("{:#?}", slopes);
 
    let result = slopes.get(&1).unwrap().calc_times_to_plow(&slopes);
    println!("{result}");
 
    Ok(())
}

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 1, 2

Verdict: ACCEPTED

Test 3

Group: 1, 2

Verdict: ACCEPTED

Test 4

Group: 2

Verdict: ACCEPTED

Test 5

Group: 2

Verdict: ACCEPTED

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: ACCEPTED

Test 11

Group: 2

Verdict: ACCEPTED

Test 12

Group: 1, 2

Verdict: ACCEPTED

Test 13

Group: 2

Verdict: ACCEPTED