开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第31天,点击查看活动详情
一、题目描述:
993. 二叉树的堂兄弟节点 - 力扣(LeetCode) (leetcode-cn.com)
在二叉树中,根节点位于深度 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
提示:
- 二叉树的节点数介于 2 到 100 之间。
- 每个节点的值都是唯一的、范围为 1 到 100 的整数。
二、思路分析:
本题可以深搜也可以广搜,这里给出一种bfs代码的实现思路。基本代码套用bfs模板即可,细节上需要注意:拿到每层的二叉树结点时,设置两个标志位用于记录该层的节点值是否与x和y匹配,如果x和y都在本层中找到,则返回true。但是需要排除x和y是同一个节点的左右节点这种情况,可以在向队列中添加新节点的时候进行排除,具体做法是设置两个数numl和numr用于记录当前节点的左右节点的值(如果它们存在的话),如果这两个值正好与x和y相等(注意与顺序无关),那么x和y所在的节点就不是堂兄弟节点,返回false即可。
三、AC 代码:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public boolean isCousins(TreeNode root, int x, int y) {
if(root==null) return false;
Deque<TreeNode> deque=new ArrayDeque<>();
deque.addLast(root);
while(!deque.isEmpty()){
int size=deque.size();
boolean tagx=false;
boolean tagy=false;
for(int i=0;i<size;++i){
int numl=-1;
int numr=-1;
TreeNode node=deque.getFirst();
if(x==node.val) tagx=true;
if(y==node.val) tagy=true;
//如果x和y都在同一层中找到,则返回true
if(tagx&&tagy) return true;
if(node.left!=null) {
deque.addLast(node.left);
numl=node.left.val;
}
if(node.right!=null) {
deque.addLast(node.right);
numr=node.right.val;
}
//排除x和y是同一个节点的左右节点的情况
if(numl==x&&numr==y) return false;
if(numr==x&&numl==y) return false;
deque.removeFirst();
}
}
return false;
}
}