【js】Leetcode每日一题-二叉树的堂兄弟节点
【题目描述】
在二叉树中,根节点位于深度 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
提示:
1.二叉树的节点数介于 2 到 100 之间。
2.每个节点的值都是唯一的、范围为 1 到 100 的整数。
【分析】
二叉树dfs | bfs遍历
代码:
/**
* 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 dx = 0, px = null, fx = false;
let dy = 0, py = null, fy = false;
const dfs = function(root, depth, parent){
if(!root) return;
if(root.val == x){
[dx, px, fx] = [depth, parent, true]
}else if(root.val == y){
[dy, py, fy] = [depth, parent, true]
}
if(fx && fy) return;
dfs(root.left, depth+1, root);
if(fx && fy) return;
dfs(root.right, depth+1, root);
}
dfs(root, 0, null);
return dx == dy && px != py;
};