P5533 [CCO 2019] Winter Driving

题目描述

在雪白帝国背部分布有$N$个城市,从$1$到$N$编号。对于一座城市$i$,有$A_i$位居民居住在其中。 它们间分布有$N-1$条道路,从$2$到$N$编号。对于一条路$j$,它连接城市$j$和$P_j$,其中$P_j \lt j$。任何城市最多被连接到36条路。 在冬季,危险的驾驶条件使得政府执行交通管制,所有的路都变为单向高速公路,即对于一条路$j$,它要么成为从城市$j$到城市$P_j$的单向高速公路,要么成为从城市$P_j$到城市$j$的单向高速公路。 此时,每个居民都想给其他所有公民寄假日贺卡。然而,如果要从城市$X$向城市$Y$发送一张假日贺卡,必须保证可以通过高速公路连接$X$和$Y$,或者$X=Y$。 请求出,当所有的公路都被转换为单向高速公路后,最多可以被送出的贺卡总数。

输入格式

第一行,一个正整数$N$。 第二行,$N$个正整数$A_1$ ~ $A_N$。 第三行,$N-1$个正整数$P_2$ ~ $P_N$。

输出格式

一行,一个正整数,即最多可以被送出的贺卡总数。

说明/提示

### 样例解释 一种可能的做法是将: ![avatar](https://s2.ax1x.com/2019/08/29/mq4eat.png) 转化为: ![avatar](https://s2.ax1x.com/2019/08/29/mq5jtx.png) 此时,3号城的每个居民都可以往3号城寄3张卡片,往2号城寄3张,往1号城寄3张,往4号城寄1张。所以3号城共计可以发送40张。 相似地,2号城共可寄18张,3号城共可寄9张,4号城一张也寄不了。 所以答案为$40+18+9+0=67$。 ### 限制 $2 \le N \le 200 000$ 对于任意合法$i$,$1 \le A_i \le 10 000$ 对于任意合法$j$,$1 \le P_j \le j$ 设$D$为任意一个城市连接的道路数量。$D \le 36$。 ### 子任务 Subtask 1(20分):$N \le 10$ Subtask 2(20分):$N \le 1000$,$D \le 10$ Subtask 3(20分):$D \le 18$ Subtask 4(20分):$N=37$,且所有的道路均连向一个城市,亦即这个城市有36条连接的城市,且它们均只通过一条道路连向此城。 Subtask 5(20分):没有附加限制。