在二叉树中,根节点位于深度 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 相等,记录层级和父节点,最后判断层级相等且不是同一个父节点为堂兄弟节点

  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. let xl
  17. let yl
  18. let p1
  19. let p2
  20. const run = (node, l) => {
  21. if (node.left && node.left.val === x) {
  22. xl = l + 1
  23. p1 = node
  24. }
  25. if (node.left && node.left.val === y) {
  26. yl = l + 1
  27. p2 = node
  28. }
  29. if (node.right && node.right.val === x) {
  30. xl = l + 1
  31. p1 = node
  32. }
  33. if (node.right && node.right.val === y) {
  34. yl = l + 1
  35. p2 = node
  36. }
  37. if (node.left) run(node.left, l + 1)
  38. if (node.right) run(node.right, l + 1)
  39. }
  40. run(root, 0)
  41. if (xl && yl && xl === yl && p1 !== p2) {
  42. return true
  43. }
  44. return false
  45. };

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