HOT993. 二叉树的堂兄弟节点

在二叉树中，根节点位于深度 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