在二叉树中,根节点位于深度 0 处,每个深度为 k 的节点的子节点位于深度 k+1 处。
如果二叉树的两个节点深度相同,但父节点不同,则它们是一对堂兄弟节点。
我们给出了具有唯一值的二叉树的根节点 root,以及树中两个不同节点的值 x 和 y。
只有与值 x 和 y 对应的节点是堂兄弟节点时,才返回 true。否则,返回 false。
实例
输入:root = [1,2,3,4], x = 4, y = 3
输出:false
/*
* @lc app=leetcode.cn id=993 lang=javascript
*
* [993] 二叉树的堂兄弟节点
*/
// @lc code=start
/**
* 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) {
// 使用map存储节点信息
const depth = new Map();
const parent = new Map();
dfs(root, null, depth, parent);
// 判断结果输出
return (depth.get(x) == depth.get(y) && parent.get(x) !== parent.get(y)) ? true : false
};
// 用来记录每个节点的父节点和深度
function dfs(root, par, depth, parent) {
// 终止条件
if (!root) return;
// 在有root的情况下
// 记录这个节点的深度,如果有父节点就使用父节点的深度+1;没有则取0,如根节点
depth.set(root.val, par ? depth.get(par.val) + 1 : 0 );
// 记录这个节点的父节点
parent.set(root.val, par? par.val : null)
// 分别遍历该节点的左右子树
dfs(root.left, root, depth, parent);
dfs(root.right, root, depth, parent);
}
// @lc code=end