本题为2024年4月13日美团机考原题
美团机考的介绍点击这里
给定一个包含 n
个节点的树,树的节点有两种颜色:红色(R)和黑色(B)。树的根节点为1号节点。每个节点的颜色由字符串 s
给出,s[i]
为 B
表示该节点颜色为黑色,为 R
则表示该节点颜色为红色。接下来,给定树的边,表示树的结构。
要求我们计算并输出有多少个子树中,既有红色节点(R)也有黑色节点(B)。
使用我们之前所学的深度优先搜索(DFS) 来遍历每个节点的子树,判断子树中是否存在红色和黑色的节点。