/**
* @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])
}
};
评论