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

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

层次遍历队列实现(非完全二叉树也可以用这种方法)
在这里插入图片描述

class Solution {
    public Node connect(Node root) {
        if (root == null) {
            return root;
        }
        
        // 初始化队列同时将第一层节点加入队列中,即根节点
        Queue<Node> queue = new LinkedList<Node>(); 
        queue.add(root);
        
        // 外层的 while 循环迭代的是层数
        while (!queue.isEmpty()) {
            
            // 记录当前队列大小
            int size = queue.size();
            
            // 遍历这一层的所有节点
            for (int i = 0; i < size; i++) {
                
                // 从队首取出元素
                Node node = queue.poll();
                
                // 连接
                if (i < size - 1) {
                    node.next = queue.peek();
                }
                
                // 拓展下一层节点
                if (node.left != null) {
                    queue.add(node.left);
                }
                if (node.right != null) {
                    queue.add(node.right);
                }
            }
        }
        
        // 返回根节点
        return root;
    }
}

将每一层的点都连接起来。每个结点的next指针都指向右边的结点,这样分两种情况:
对root的下一层进行连接(通过root的next指针)
如果右边结点是亲兄弟结点,则在遍历到父节点的时候就root->left->next = root->right;;
如果右边结点是堂兄弟结点,则在遍历到父节点的时候(此时父节点的next指针已经指向了右边的兄弟节点),通过父节点的next指针与右边结点相接: root->right->next = root->next->left;

class Solution {
public:
    Node* connect(Node* root) {
        if(root==NULL){
            return NULL;
        }
        if(root->left!=NULL){//左子树
            root->left->next=root->right;
        }
        if(root->right!=NULL&&root->next!=NULL){//右子树(用root的next来定位
                                                //右子树的next)
            root->right->next=root->next->left;
        }
        root->left=connect(root->left);
        root->right=connect(root->right);
        return root;
        
    }
};
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值