LeetCode题解:993. 二叉树的堂兄弟节点,BFS,Ja

2023-11-01  本文已影响0人  Lee_Chen

原题链接:
https://leetcode.cn/problems/cousins-in-binary-tree/

解题思路:

  1. 使用队列进行BFS搜索,同时保存每个节点,以及其深度和父节点信息。
  2. 当搜索到xy时,对比深度和父节点,如果满足要求,则表示找到了堂兄弟节点。
/**
 * @param {TreeNode} root
 * @param {number} x
 * @param {number} y
 * @return {boolean}
 */
var isCousins = function (root, x, y) {
  // 使用队列进行BFS搜索,每个元素保存的值是当前节点、节点深度、父节点
  let queue = [[root, 1, null]]
  // 保存搜索到的x和y节点信息
  let result = []

  // 不断搜索直到队列被清空,表示完成了对二叉树的搜索。
  while (queue.length) {
    // 将队列元素出队,获取相关信息
    const [node, depth, parent] = queue.shift()

    // 当查找到x或y的值时,将相应的信息保存到result
    if (node.val === x || node.val === y) {
      result.push([node, depth, parent])
    }

    // 如果result的长度为2,表示已查找到x和y
    if (result.length === 2) {
      // 如果x和y的深度相等,父节点不同,表示找到了堂兄弟节点
      if (result[0][1] === result[1][1] && result[0][2] !== result[1][2]) {
        return true
      }

      return false
    }

    // 将当前节点的左右子节点入队,继续搜索
    node.left && queue.push([node.left, depth + 1, node])
    node.right && queue.push([node.right, depth + 1, node])
  }
};
上一篇 下一篇

猜你喜欢

热点阅读