每日一题 leetcode 993.二叉树的堂兄弟结点

题:
在这里插入图片描述


思:
用第二个队列来保存在同一层的所有结点,
如果能找到x和y,且它们不拥有同一个父节点,则返回true
如果只能找到一个x或者y,则返回false,因为此时这意味着另一个x或者y,和我们这一层找到的x或者y一定不在同一层
如果既没找到x又没找到y,则再找下一层
最后也没找到,返回false


码:

public boolean isCousins(TreeNode root, int x, int y) {
        if (root.val == x || root.val == y)
            return false;
        // 如果深度相同,且不是同一个父亲,则返回true
        // 当深度为h的时候,应该弹出2的h次方个结点
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        Queue<TreeNode> tempQueue = new LinkedList<TreeNode>();
        queue.offer(root);
        int count = 0;
        while (!queue.isEmpty()) {
            // 弹出所有的元素,此时弹出的元素都在同一层
            while (!queue.isEmpty()) {
                TreeNode poll = queue.poll();
                tempQueue.offer(poll);
            }
            // 分别将这些元素的孩子加入队列queue,如果两个孩子正好是x和y,返回false
            // 如果所有的孩子加完以后,只有一个x或者y,则返回false
            // 如果能找到不同父亲的两个孩子,则返回true
            while (!tempQueue.isEmpty()) {
                TreeNode poll = tempQueue.poll();
                if (poll.left != null && poll.right != null) {
                    if ((poll.left.val == x && poll.right.val == y) || (poll.left.val == y && poll.right.val == x))
                        return false;
                    queue.offer(poll.left);
                    if (poll.left.val == x || poll.left.val == y) count++;
                    queue.offer(poll.right);
                    if (poll.right.val == x || poll.right.val == y) count++;
                } else if (poll.left != null){
                    queue.offer(poll.left);
                    if (poll.left.val == x || poll.left.val == y) count++;
                }else if(poll.right != null){
                    queue.offer(poll.right);
                    if (poll.right.val == x || poll.right.val == y) count++;
                }
            }
            if(count == 1)return false;
            if(count == 2)return true;
        }
        return false;
    }
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值