在二叉树中,根节点位于深度 0 处,每个深度为 k 的节点的子节点位于深度 k+1 处。
如果二叉树的两个节点深度相同,但 父节点不同 ,则它们是一对堂兄弟节点。
我们给出了具有唯一值的二叉树的根节点 root ,以及树中两个不同节点的值 x 和 y 。
只有与值 x 和 y 对应的节点是堂兄弟节点时,才返回 true 。否则,返回 false。
示例
示例 1:
输入:root = [1,2,3,4], x = 4, y = 3
输出:false
示例 2:
输入:root = [1,2,3,null,4,null,5], x = 5, y = 4
输出:true
示例 3:
输入:root = [1,2,3,null,4], x = 2, y = 3
输出:false
题解
前序遍历二叉树,判断节点值是否与 x、y 相等,记录层级和父节点,最后判断层级相等且不是同一个父节点为堂兄弟节点
/**
* Definition for a binary tree node.
* function TreeNode(val, left, right) {
* this.val = (val===undefined ? 0 : val)
* this.left = (left===undefined ? null : left)
* this.right = (right===undefined ? null : right)
* }
*/
/**
* @param {TreeNode} root
* @param {number} x
* @param {number} y
* @return {boolean}
*/
var isCousins = function(root, x, y) {
let xl
let yl
let p1
let p2
const run = (node, l) => {
if (node.left && node.left.val === x) {
xl = l + 1
p1 = node
}
if (node.left && node.left.val === y) {
yl = l + 1
p2 = node
}
if (node.right && node.right.val === x) {
xl = l + 1
p1 = node
}
if (node.right && node.right.val === y) {
yl = l + 1
p2 = node
}
if (node.left) run(node.left, l + 1)
if (node.right) run(node.right, l + 1)
}
run(root, 0)
if (xl && yl && xl === yl && p1 !== p2) {
return true
}
return false
};
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/cousins-in-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。