AT_tkppc3_i.王国と M 種類の店
通过率:0%
AC君温馨提醒
该题目为【atcoder】题库的题目,您提交的代码将被提交至atcoder进行远程评测,并由ACGO抓取测评结果后进行展示。由于远程测评的测评机由其他平台提供,我们无法保证该服务的稳定性,若提交后无反应,请等待一段时间后再进行重试。
题目描述
PAKEN 王国是一片由 N 个顶点构成的树状结构。树中的每个顶点 i(2≤i≤N)与顶点 Pi 通过一条长度为 Li 的边相连。
在这个王国中,有 M 种不同类型的商店,比如文具店、超市等(我们将它们称作为种类 1、种类 2、至种类 M)。每个顶点上恰好有一家商店,且顶点 i 上的商店类型为 Ri。
对于每个顶点 i,我们定义其「不便度」为从顶点 i 出发到达最近的各类型商店的最短距离总和,也就是到最近的种类 1 的商店的距离、加上到种类 2 的商店的距离,直至加上到种类 M 的商店的最短距离。
你的任务是计算出顶点 1、顶点 2、一直到顶点 N 这每一个顶点的「不便度」。
输入格式
输入包含以下内容,以标准输入形式给出:
- 第一行两个整数 N 和 M,分别表示顶点数和商店种类数。
- 第二行包含 N−1 个整数,依次为 P2,P3,…,PN,表示边的起点。
- 第三行包含 N−1 个整数,依次为 L2,L3,…,LN,表示边的长度。
- 第四行包含 N 个整数,依次为 R1,R2,R3,…,RN,表示各顶点上的商店类型。
输出格式
共输出 N 行,每行一个整数,表示对应顶点的「不便度」。
输入输出样例
输入#1
5 2 1 2 1 2 3 5 2 4 2 1 2 1 1
输出#1
2 3 5 2 7
输入#2
10 3 1 2 3 3 1 6 7 8 8 2 3 5 3 1 1 6 2 7 1 3 2 2 3 2 3 1 1 3
输出#2
3 5 8 18 11 2 3 13 17 21
输入#3
5 5 1 2 3 4 1 2 3 4 1 2 3 4 5
输出#3
20 17 15 18 30
说明/提示
注意
每个测试用例的输入数据量较大,可能达到 7.5 MB,因此在 C++ 中建议使用 scanf 和 printf 以替代 cin 和 cout。
制约
- 1≤N≤400,000
- 1≤M≤400,000
- 1≤Pi≤i−1
- 1≤Li≤1,000,000
- 1≤Ri≤M
- 每种商店类型至少存在一个。
子任务
- 子任务 1 [ 55 分 ]: N≤1,500,M≤750。
- 子任务 2 [ 110 分 ]: N,M≤70,000 且 N=M。
- 子任务 3 [ 275 分 ]: N,M≤70,000 且每种商店类型最大仅有 8 个。
- 子任务 4 [ 330 分 ]: N,M≤70,000。
- 子任务 5 [ 330 分 ]: N,M≤400,000。
本翻译由 AI 自动生成
输入解题思路,AI测评打分。不知道怎么写?