leetCode二叉树的堂兄弟节点

题目描述

在二叉树中，根节点位于深度 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
993. 二叉树的堂兄弟节点

解答思路

本题是一道简单题，本身的测试用例较为简单，本身思路也比较清晰，主要依靠bfs进行搜索，因为是求解堂兄弟，所以我们使用bfs可以很快的求解同一层的兄弟，再遍历过程中是事先判断两个值是否是亲兄弟，而后遍历过程中按层次遍历，如果出现目标数字就将+1，如果出现s==1说明不在同一层，s==2说明成功找到即可返回true。

代码如下

class Solution {
    public boolean isCousins(TreeNode root, int x, int y) {
        return bfs(root,x,y);
    }
    public boolean bfs(TreeNode root,int x,int y){
            Queue<TreeNode> q=new LinkedList<TreeNode>();
            if(root.val==x||root.val==y)//首节点不可能有堂兄弟
                return false;
            q.add(root);
            while(!q.isEmpty()){
                int s=0;
                int n=q.size();//判断层次
                while(n--!=0){
                    TreeNode t=q.poll();
                    if(t.left!=null)
                        q.add(t.left);
                    if(t.right!=null)
                        q.add(t.right);
                    if((t.left!=null&&t.right!=null)&&(t.left.val==x&&t.right.val==y||t.left.val==y&&t.right.val==x)){//事先判断
                        return false;
                    }
                    if(t.val==x||t.val==y){//标记
                        s++;
                    }
                }
                if(s==1)
                    return false;
                else if(s==2)
                    return true;
            }
            return false;
    }
}