填空题

黄黑树

发布于 2022-03-03 16:16:05

小团作为一名美团骑手,最喜欢的颜色就是黄和黑,因此对包含这两种颜色的事物都格外留意。

这天他发现有一棵树,树上的每个节点都是黄的或者黑的。现在小团想知道对于这棵树中的每个节点,在以其为根的子树中,所有与其颜色不同的节点的深度之和是多少。子树中节点的深度被定义为该节点与该子树根节点之间的最短路径的边数。



输入描述:

第一行有一个数n,代表这棵树上一共有n个节点,编号为1~n。(1&ltn<=100000)

第二行有n个数,第i个数表示编号为i的节点的颜色,为0表示黄色,为1表示黑色。

第三行有n-1个数,第i个数表示编号为i+1的节点的父节点编号。

输入样例: 10 0 0 1 0 0 1 1 1 0 0 1 2 3 4 4 5 7 6 9 输出描述: 在一行中输出n个数,第i个数代表第i个节点的答案。输出样例 17 13 10 6 3 3 0 0 0 0
关注者
0
被浏览
18
知识点
面圈网VIP题库

面圈网VIP题库全新上线,海量真题题库资源。 90大类考试,超10万份考试真题开放下载啦

去下载看看