#P5533. [CCO2019] Winter Driving

[CCO2019] Winter Driving

题目描述

在雪白帝国背部分布有NN个城市,从11NN编号。对于一座城市ii,有AiA_i位居民居住在其中。

它们间分布有N1N-1条道路,从22NN编号。对于一条路jj,它连接城市jjPjP_j,其中Pj<jP_j \lt j。任何城市最多被连接到36条路。

在冬季,危险的驾驶条件使得政府执行交通管制,所有的路都变为单向高速公路,即对于一条路jj,它要么成为从城市jj到城市PjP_j的单向高速公路,要么成为从城市PjP_j到城市jj的单向高速公路。

此时,每个居民都想给其他所有公民寄假日贺卡。然而,如果要从城市XX向城市YY发送一张假日贺卡,必须保证可以通过高速公路连接XXYY,或者X=YX=Y

请求出,当所有的公路都被转换为单向高速公路后,最多可以被送出的贺卡总数。

输入格式

第一行,一个正整数NN

第二行,NN个正整数A1A_1 ~ ANA_N

第三行,N1N-1个正整数P2P_2 ~ PNP_N

输出格式

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

4
3 3 4 1
1 2 1
67

提示

样例解释

一种可能的做法是将:

avatar

转化为:

avatar

此时,3号城的每个居民都可以往3号城寄3张卡片,往2号城寄3张,往1号城寄3张,往4号城寄1张。所以3号城共计可以发送40张。

相似地,2号城共可寄18张,3号城共可寄9张,4号城一张也寄不了。

所以答案为40+18+9+0=6740+18+9+0=67

限制

2N2000002 \le N \le 200 000

对于任意合法ii1Ai100001 \le A_i \le 10 000

对于任意合法jj1Pjj1 \le P_j \le j

DD为任意一个城市连接的道路数量。D36D \le 36

子任务

Subtask 1(20分):N10N \le 10

Subtask 2(20分):N1000N \le 1000D10D \le 10

Subtask 3(20分):D18D \le 18

Subtask 4(20分):N=37N=37,且所有的道路均连向一个城市,亦即这个城市有36条连接的城市,且它们均只通过一条道路连向此城。

Subtask 5(20分):没有附加限制。