COCI '23 Contest 3 #5 Slučajna Cesta

View as PDF

Submit solution

Points: 15 (partial)
Time limit: 3.0s
Memory limit: 512M

Problem types

Vito lives in a city with n parks labeled from 1 to n. The parks are connected with n-1 roads such that there is a path between any two pairs of parks. Every park has some beauty value; the beauty value of the i-th park is v_i.

Last night Vito decided to wander around the city in such a way that after he visits a park he chooses a random road with equal probability and visits a park to which that road leads. But before he started his journey he looked through the window of his skyscraper and saw that on every road there is either a blue or a red snake. Blue snakes attack all people traveling from the park with a lower label to a park with a higher one, and red snakes attack everyone traveling from a park with a higher label to a park with a lower one. As Vito doesn't want to get attacked by a snake he decided to change his plans by considering only roads on which he will not get attacked by a snake when choosing a random road. Since he likes long walks he will not stop on his journey until there is at least one road he can safely pass.

And while Vito walks down the stairs of his skyscraper he completely forgot on which road is red or blue snake so he wonders: If on every road there is an equal probability of a blue or a red snake, what is the expected beauty of my journey which starts in the i-th park?

The beauty of a path is the sum of beauties of parks visited on that journey. The expected beauty of a journey is defined as the sum of the product of the beauty of a path and the probability Vito takes that path, for every possible path.

Input Specification

In the first line there is an integer n (2 \le n \le 10^6), which denotes the number of parks.

In the second line there are n-1 integers p_i (1 \le p_i < i), which denote a road between the (i+1)-th park and p_i-th park.

In the third line there are n integers v_i (0 \le v_i \le 10^6), where v_i denotes the beauty of the i-th park.

Output Specification

If the expected beauty of Vito's journey which starts at the i-th park is \frac{a}{b} for integers a and b, then in the i-th line of output, print ab^{-1} \pmod{10^9 + 7} where b^{-1} is modular inverse of b \pmod{10^9 + 7}.

Scoring

Subtask Points Constraints
1 10 n \le 10
2 30 n \le 1\,000
3 30 In the sequence p_i, no value is present more than 2 times.
4 40 No additional constraints.

If your program, on some test, outputs the first line correct, but outputs a wrong answer in the following lines, it will receive 50\% of the points for that test.

The number of points in a subtask corresponds to the least number of points achieved by some test in that subtask.

Sample Input 1

2
1
2 1

Sample Output 1

500000006
2

Explanation for Sample 1

The expected beauty of a journey starting at the first park is 2.5 \pmod{10^9 + 7} = \frac{5}{2} \pmod{10^9 + 7} = 5 \cdot 2^{-1} \pmod{10^9 + 7} = 5 \cdot 500000004 \pmod{10^9 + 7} = 500000006 \pmod{10^9 + 7} and starting from the second park it is 2.

Sample Input 2

3
1 1
8 8 8

Sample Output 2

14
14
14

Explanation for Sample 2

The probability that both snakes are red is \frac{1}{4} and in that case if Vito starts at the first park he randomly chooses which road he will take.

Sample Input 3

11
1 1 1 2 3 4 1 2 6 2
1 1000 5 3 18 200 8 9 0 2 2

Sample Output 3

968750272
610352580
450521029
536458466
199219275
662760680
190972315
90277951
824219264
941840425
532552597

Comments

There are no comments at the moment.