#39. 腾讯2024春季-1.小红的二叉树路径权值
题目描述
题解
题库
腾讯2024春季-1.小红的二叉树路径权值
题目描述
小红拿到了一棵二叉树,每个节点的权值为0或1。
小红希望从根节点出发,每次前往左儿子或者右儿子,直到走到叶子时停止,路径上所有节点将表示为一个二进制数(按路径顺序)。小红希望最终的二进制数在区间[l, r]内,请你告诉小红有多少种不同方案。
输入描述
输入包含n + 2行,第一行输入一个整数n,表示二叉树的节点数量;
第二行输入n个整数,表示每个节点的权值ai;
后续n - 1行,每行输入两个整数u, v,表示编号为u,v节点连接了一条边。
最后一行输入两个整数,表示l, r.
1 <= n < = 105;
0 <= ai <= 1;
1 <= u, v <= n;
1 <= l <= r <= 109.
题目保证根节点的编号为1。
输出描述
输出一个整数表示答案。
输入示例
4
1 0 1 1
1 2
1 3
2 4
2 3
输出示例
1
提示信息
选择路径1 -> 3,得到的二进制数字是11,表示十进制的3,在[2, 3]范围内。即一条路径满足题目所述条件。
时间限制:c/c++:1s;其他语言:4s。