返回首页 返回题库

二叉树

C++ 中等 GESP六级真题 #二叉树#深度优先搜索#差分

题目描述

题面描述

小杨有一棵包含 个节点的二叉树,且根节点的编号为1。这棵二叉树任意一个节点要么是白色,要么是黑色。之后小杨会对这棵二叉树进行 次操作,每次小杨会选择一个节点,将以这个节点为根的子树内所有节点的颜色反转,即黑色变成白色,白色变成黑色。

小杨想知道 次操作全部完成之后每个节点的颜色。

输入格式

第一行一个正整数 ,表示二叉树的节点数量。

第二行 个正整数,第 ()个数表示编号为 的节点的父亲节点编号,数据保证是一棵二叉树。

第三行一个长度为 的01串,从左到右第 ()位如果为0,表示编号为 的节点颜色为白色,否则为黑色。

第四行一个正整数 ,表示操作次数。

接下来 行每行一个正整数 (),表示第 次操作选择的节点编号。

输出格式

输出一行一个长度为 的01串,表示 次操作全部完成之后每个节点的颜色。从左到右第 ()位如果为0,表示编号为 的节点颜色为白色,否则为黑色。

样例1

输入:

6
3 1 1 3 4
100101
3
1
3
2

输出:

010000

样例解释

第一次操作后,节点颜色为:011010

第二次操作后,节点颜色为:000000

第三次操作后,节点颜色为:010000

数据范围

对于全部数据,保证有 。

时间限制

1.0 s

内存限制

512.0 MB

💡 查看提示(先自己想一想再看)
本题考察二叉树的深度优先遍历与差分思想。不需要每次操作都遍历整个子树,只需要记录每个节点被翻转的次数,最后通过一次DFS统计从根到每个节点的总翻转次数,判断是否需要反转颜色即可,时间复杂度为O(n+q),可以处理1e5的数据规模。
⚙ 开始做这道题 0 人做过 · 0 收藏 · 2 浏览

讨论(0)