Leetcode 116:填充每一个节点的下一个右侧节点指针

给定一个 完美二叉树 ,其所有叶子节点都在同一层,每个父节点都有两个子节点。二叉树定义如下:

struct Node {
  int val;
  Node *left;
  Node *right;
  Node *next;
}

填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL

初始状态下,所有 next 指针都被设置为 NULL

思路:层次遍历的时候,使用next节点指向当前节点的下一个节点。

连续从对列中弹出两个节点,当前节点的next就是第二个从队列中弹出的元素。

public Node connect(Node root) {
        Queue<Node> queue=new LinkedList<>();
        if(root!=null) queue.add(root);

        while (!queue.isEmpty()){
            int size=queue.size();
            Node node=queue.poll();
            if(node.left!=null) queue.add(node.left);
            if(node.right!=null) queue.add(node.right);

            for(int i=1;i<size;i++){
                Node next=queue.poll();
                if(next.left!=null) queue.add(next.left);
                if(next.right!=null) queue.add(next.right);
                node.next=next;
                node=next;
            }
        }
        return root;
    }

  • 3
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值