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