每日一题:993. 二叉树的堂兄弟节点

64 阅读2分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第31天,点击查看活动详情

一、题目描述:

993. 二叉树的堂兄弟节点 - 力扣(LeetCode) (leetcode-cn.com)

在二叉树中,根节点位于深度 0 处,每个深度为 k 的节点的子节点位于深度 k+1 处。

如果二叉树的两个节点深度相同,但 父节点不同 ,则它们是一对堂兄弟节点。

我们给出了具有唯一值的二叉树的根节点 root ,以及树中两个不同节点的值 x 和 y 。

只有与值 x 和 y 对应的节点是堂兄弟节点时,才返回 true 。否则,返回 false。

 

示例 1:

image.png

输入:root = [1,2,3,4], x = 4, y = 3
输出:false

示例 2:

image.png

输入:root = [1,2,3,null,4,null,5], x = 5, y = 4
输出:true

示例 3:

image.png

输入: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;
    }
}