【每日一题Day367】LC117填充每个节点的下一个右侧节点指针II | BFS

117.填充每个节点的下一个右侧节点指针II

给定一个二叉树:

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

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

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

与LC116的区别在于,这个二叉树可能不是完全二叉树

  • BFS:使用队列进行BFS

    java 复制代码
    /*
    // Definition for a Node.
    class Node {
        public int val;
        public Node left;
        public Node right;
        public Node next;
    
        public Node() {}
        
        public Node(int _val) {
            val = _val;
        }
    
        public Node(int _val, Node _left, Node _right, Node _next) {
            val = _val;
            left = _left;
            right = _right;
            next = _next;
        }
    };
    */
    
    class Solution {
     public Node connect(Node root) {
            if (root == null)
                return root;
            //cur我们可以把它看做是每一层的链表
            Node cur = root;
            while (cur != null) {
                //遍历当前层的时候,为了方便操作在下一
                //层前面添加一个哑结点(注意这里是访问
                //当前层的节点,然后把下一层的节点串起来)
                Node dummy = new Node(0);
                //pre表示访下一层节点的前一个节点
                Node pre = dummy;
                //然后开始遍历当前层的链表
                while (cur != null) {
                    if (cur.left != null) {
                        //如果当前节点的左子节点不为空,就让pre节点
                        //的next指向他,也就是把它串起来
                        pre.next = cur.left;
                        //然后再更新pre
                        pre = pre.next;
                    }
                    //同理参照左子树
                    if (cur.right != null) {
                        pre.next = cur.right;
                        pre = pre.next;
                    }
                    //继续访问这样行的下一个节点
                    cur = cur.next;
                }
                //把下一层串联成一个链表之后,让他赋值给cur,
                //后续继续循环,直到cur为空为止
                cur = dummy.next;
            }
            return root;
        }
    }
  • 迭代遍历

    1. 设置节点cur,遍历每一层的链表
    2. 设置节点pre,将下一层的节点串起来
    • 时间复杂度:O(n)
    • 空间复杂度:O(1)
    java 复制代码
    /*
    // Definition for a Node.
    class Node {
        public int val;
        public Node left;
        public Node right;
        public Node next;
    
        public Node() {}
        
        public Node(int _val) {
            val = _val;
        }
    
        public Node(int _val, Node _left, Node _right, Node _next) {
            val = _val;
            left = _left;
            right = _right;
            next = _next;
        }
    };
    */
    
    class Solution {
     public Node connect(Node root) {
            if (root == null)
                return root;
            //cur我们可以把它看做是每一层的链表
            Node cur = root;
            while (cur != null) {
                //遍历当前层的时候,为了方便操作在下一
                //层前面添加一个哑结点(注意这里是访问
                //当前层的节点,然后把下一层的节点串起来)
                Node dummy = new Node(0);
                //pre表示访下一层节点的前一个节点
                Node pre = dummy;
                //然后开始遍历当前层的链表
                while (cur != null) {
                    if (cur.left != null) {
                        //如果当前节点的左子节点不为空,就让pre节点
                        //的next指向他,也就是把它串起来
                        pre.next = cur.left;
                        //然后再更新pre
                        pre = pre.next;
                    }
                    //同理参照左子树
                    if (cur.right != null) {
                        pre.next = cur.right;
                        pre = pre.next;
                    }
                    //继续访问这样行的下一个节点
                    cur = cur.next;
                }
                //把下一层串联成一个链表之后,让他赋值给cur,
                //后续继续循环,直到cur为空为止
                cur = dummy.next;
            }
            return root;
        }
    }
相关推荐
爪哇学长20 分钟前
双指针算法详解:原理、应用场景及代码示例
java·数据结构·算法
Dola_Pan24 分钟前
C语言:数组转换指针的时机
c语言·开发语言·算法
繁依Fanyi37 分钟前
简易安卓句分器实现
java·服务器·开发语言·算法·eclipse
烦躁的大鼻嘎1 小时前
模拟算法实例讲解:从理论到实践的编程之旅
数据结构·c++·算法·leetcode
C++忠实粉丝1 小时前
计算机网络socket编程(4)_TCP socket API 详解
网络·数据结构·c++·网络协议·tcp/ip·计算机网络·算法
用户37791362947552 小时前
【循环神经网络】只会Python,也能让AI写出周杰伦风格的歌词
人工智能·算法
福大大架构师每日一题2 小时前
文心一言 VS 讯飞星火 VS chatgpt (396)-- 算法导论25.2 1题
算法·文心一言
EterNity_TiMe_2 小时前
【论文复现】(CLIP)文本也能和图像配对
python·学习·算法·性能优化·数据分析·clip
机器学习之心2 小时前
一区北方苍鹰算法优化+创新改进Transformer!NGO-Transformer-LSTM多变量回归预测
算法·lstm·transformer·北方苍鹰算法优化·多变量回归预测·ngo-transformer
yyt_cdeyyds2 小时前
FIFO和LRU算法实现操作系统中主存管理
算法