来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/cousins-in-binary-tree 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

在二叉树中,根节点位于深度 0 处,每个深度为 k 的节点的子节点位于深度 k+1 处。

如果二叉树的两个节点深度相同,但 父节点不同 ,则它们是一对堂兄弟节点。

我们给出了具有唯一值的二叉树的根节点 root ,以及树中两个不同节点的值 x 和 y 。

只有与值 x 和 y 对应的节点是堂兄弟节点时,才返回 true 。否则,返回 false。

解答

从树中找到目标节点,并返回它的父节点和深度

  1. /**
  2. * Definition for a binary tree node.
  3. * function TreeNode(val, left, right) {
  4. * this.val = (val===undefined ? 0 : val)
  5. * this.left = (left===undefined ? null : left)
  6. * this.right = (right===undefined ? null : right)
  7. * }
  8. */
  9. /**
  10. * @param {TreeNode} root
  11. * @param {number} x
  12. * @param {number} y
  13. * @return {boolean}
  14. */
  15. var isCousins = function(root, x, y) {
  16. function traverse (node, target, parent, deep) {
  17. if (node === null) {
  18. return null;
  19. }
  20. if (node.val === target) {
  21. return { parent, deep };
  22. }
  23. return traverse(node.left, target, node, deep + 1) || traverse(node.right, target, node, deep + 1);
  24. }
  25. const xObj = traverse(root, x, null, 0);
  26. const yObj = traverse(root, y, null, 0);
  27. return xObj?.parent !== yObj?.parent && xObj?.deep === yObj?.deep;
  28. };