LeetCode每日一题 2021/5/17 993. 二叉树的堂兄弟节点

本文链接：https://blog.csdn.net/qq_36917571/article/details/117026987

题目描述

在二叉树中，根节点位于深度 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

思路

根据题意，对于两个节点 X 和 Y，如果它们的深度相同，且父节点不同，则 X 和 Y 是一对堂兄弟节点，否则就不是。涉及到求二叉树节点的深度，则使用层序遍历（BFS遍历）较为方便。关于层序遍历，建议先看一下二叉树的层序遍历，看这一篇就搞定了
如何求 X 和 Y 的深度以及父节点呢？采用层序遍历，记录每一层的深度。对每一个节点，如果其子节点等于 X（或者 Y），则这个节点就是 X（或者 Y ）的父节点，这个节点的深度加上一就是 X （或者 Y）的深度。最后比较 X 和 Y 的深度以及父节点是否相同

Java代码

class Solution {
    public boolean isCousins(TreeNode root, int x, int y) {
        Queue<TreeNode> que = new LinkedList<>();
        que.offer(root);
        int h = 0;                                   // 初始化当前层的深度
        int dx = 0, dy = 0;                          // 初始化 X 和 Y 的深度       
        TreeNode px = null, py = null;               // 初始化 X 和 Y 的父节点
        while(!que.isEmpty()) {
            h++;                                     // 当前层的深度
            int size = que.size();
            for(int i = 0; i< size; i++) {
                TreeNode cur = que.poll();
                if(cur.left != null) {
                    que.offer(cur.left);
                    if(cur.left.val == x) {          // 如果当前节点的左子节点等于 X，则这个节点就是 X 的父节点，这个节点的深度加上一就是 X 的深度         
                        dx = h + 1;
                        px = cur;
                    }
                    if(cur.left.val == y) {          // 如果当前节点的左子节点等于 Y，则这个节点就是 Y 的父节点，这个节点的深度加上一就是 Y 的深度         
                        dy = h + 1;
                        py = cur;
                    }
                }
                if(cur.right != null) {              
                    que.offer(cur.right);
                    if(cur.right.val == x) {         // 如果当前节点的右子节点等于 X，则这个节点就是 X 的父节点，这个节点的深度加上一就是 X 的深度         
                        dx = h + 1;
                        px = cur;
                    }
                    if(cur.right.val == y) {         // 如果当前节点的右子节点等于 Y，则这个节点就是 Y 的父节点，这个节点的深度加上一就是 Y 的深度         
                        dy = h + 1;
                        py = cur;
                    }
                }
            } 
        }
        if(dx == dy && px != py) {                   // 如果 X 和 Y 的深度相同且父节点不同，则它们是一对堂兄弟节点
            return true;
        } else {
            return false;
        }
    }
}