CSES - Datatähti 2024 alku - Results
Submission details
Task:Laskettelukeskus
Sender:Bliz
Submission time:2023-11-01 09:18:36 +0200
Language:Rust
Status:READY
Result:100
Feedback
groupverdictscore
#1ACCEPTED53
#2ACCEPTED47
Test results
testverdicttimegroup
#1ACCEPTED0.00 s1, 2details
#2ACCEPTED0.00 s1, 2details
#3ACCEPTED0.00 s1, 2details
#4ACCEPTED0.05 s2details
#5ACCEPTED0.05 s2details
#6ACCEPTED0.00 s1, 2details
#7ACCEPTED0.05 s2details
#8ACCEPTED0.00 s1, 2details
#9ACCEPTED0.04 s2details
#10ACCEPTED0.00 s1, 2details
#11ACCEPTED0.04 s2details
#12ACCEPTED0.00 s1, 2details
#13ACCEPTED0.04 s2details

Code

use std::{cmp, io};
#[derive(Clone, Debug)]
struct Slope {
plow_count: u64,
children: Vec<usize>,
}
fn main() {
let mut input = String::new();
io::stdin().read_line(&mut input).unwrap();
let n: usize = input.trim().parse().unwrap();
let mut slopes = vec![
Slope {
plow_count: 0,
children: Vec::new(),
};
n
];
build_graph(&mut slopes, n);
read_plow_counts(&mut slopes);
let result = count_paths(&slopes, 0);
println!("{}", result);
}
fn build_graph(slopes: &mut Vec<Slope>, n: usize) {
for _ in 0..n - 1 {
let mut input = String::new();
io::stdin().read_line(&mut input).unwrap();
let nums: Vec<usize> = input
.split_whitespace()
.map(|s| s.parse().unwrap())
.collect();
let parent = nums[0] - 1;
let child = nums[1] - 1;
slopes[parent].children.push(child);
}
}
fn read_plow_counts(slopes: &mut Vec<Slope>) {
let mut input = String::new();
io::stdin().read_line(&mut input).unwrap();
let nums: Vec<u64> = input
.split_whitespace()
.map(|s| s.parse().unwrap())
.collect();
for (i, num) in nums.iter().enumerate() {
slopes[i].plow_count = *num;
}
}
fn count_paths(slopes: &Vec<Slope>, index: usize) -> u64 {
let slope = &slopes[index];
if slope.children.is_empty() {
slope.plow_count
} else {
cmp::max(
slope.plow_count,
slope
.children
.iter()
.map(|&i| count_paths(&slopes, i))
.sum(),
)
}
}

Test details

Test 1

Group: 1, 2

Verdict: ACCEPTED

input
5
1 2
1 3
3 4
3 5
...

correct output
6

user output
6

Test 2

Group: 1, 2

Verdict: ACCEPTED

input
100
1 73
1 64
64 23
1 88
...

correct output
2675

user output
2675

Test 3

Group: 1, 2

Verdict: ACCEPTED

input
100
1 36
36 56
56 59
36 97
...

correct output
2808

user output
2808

Test 4

Group: 2

Verdict: ACCEPTED

input
100000
1 45452
1 74209
45452 78960
45452 79820
...

correct output
28399367694319

user output
28399367694319

Test 5

Group: 2

Verdict: ACCEPTED

input
100000
1 31165
1 23263
31165 89516
31165 53122
...

correct output
28546840313799

user output
28546840313799

Test 6

Group: 1, 2

Verdict: ACCEPTED

input
100
1 79
79 9
79 45
45 10
...

correct output
0

user output
0

Test 7

Group: 2

Verdict: ACCEPTED

input
100000
1 66038
1 56789
56789 7403
66038 69542
...

correct output
0

user output
0

Test 8

Group: 1, 2

Verdict: ACCEPTED

input
100
1 2
2 3
3 4
4 5
...

correct output
100

user output
100

Test 9

Group: 2

Verdict: ACCEPTED

input
100000
1 2
2 3
3 4
4 5
...

correct output
1000000000

user output
1000000000

Test 10

Group: 1, 2

Verdict: ACCEPTED

input
100
1 2
1 3
2 4
2 5
...

correct output
2809

user output
2809

Test 11

Group: 2

Verdict: ACCEPTED

input
100000
1 2
1 3
2 4
2 5
...

correct output
26053917212428

user output
26053917212428

Test 12

Group: 1, 2

Verdict: ACCEPTED

input
100
1 2
1 3
2 4
2 5
...

correct output
5000

user output
5000

Test 13

Group: 2

Verdict: ACCEPTED

input
100000
1 2
1 3
2 4
2 5
...

correct output
50000000000000

user output
50000000000000