leetcode.116. 填充每个节点的下一个右侧节点指针.117. 填充每个节点的下一个右侧节点指针 II.详解笔记.

文章介绍了两种方法解决二叉树节点的next指针连接问题。解法一采用层序遍历,利用队列实现,适用于所有二叉树;解法二利用完美二叉树特性,但只适用于特定情况。两种方法均对二叉树进行原地修改,无需额外空间。
摘要由CSDN通过智能技术生成

思路:

解法一:迭代+层序遍历

本解法与 leetcode.102. 二叉树的层序遍历.详解笔记._蜜桃鲜橙多的博客-CSDN博客本题考查最基本的二叉树层序遍历算法,DFS广度优先搜索,利用队列实现。学习到队列的c++表示,还有层序遍历的算法,一维和二维容器的使用。入队前都判空,可以省去很多pop语句,简便很多。https://blog.csdn.net/m0_62615295/article/details/130385351?spm=1001.2014.3001.5501

本质相同,也是考察二叉树层序遍历算法,这里用DFS实现。只不过多一步填充next。

遍历本层结点时,若当前结点右侧还有结点,也就是队列非空,且当前结点不是本层最后一个结点(这个时候之前若有结点有孩子,孩子已入队,队列非空,但不应填充next指针) ,则填充next指针。

/*
// Definition for a Node.
class Node {
public:
    int val;
    Node* left;
    Node* right;
    Node* next;

    Node() : val(0), left(NULL), right(NULL), next(NULL) {}

    Node(int _val) : val(_val), left(NULL), right(NULL), next(NULL) {}

    Node(int _val, Node* _left, Node* _right, Node* _next)
        : val(_val), left(_left), right(_right), next(_next) {}
};
*/

class Solution {
public:
    Node* connect(Node* root) {
        queue<Node*>q;
        if(root){q.push(root);}//根结点非空则入队
        while(!q.empty()){//队列非空时
        int size=q.size();
        for(int i=0;i<size;i++){//遍历本层
            Node*cur=q.front();取当前结点
            q.pop();
            if(!q.empty()&&i!=size-1){cur->next=q.front();}//当前结点右侧还有结点,填充next
            if(cur->left){q.push(cur->left);}//左右孩子非空则入队
            if(cur->right){q.push(cur->right);}
          }
        
      }
      return root; //原地改变,直接返回根结点即可
    }
};

 解法二:使用已建立的 next指针

可以充分利用完美二叉树性质,所有叶子节点都在同一层,每个父结点都有两个孩子。

从根结点开始,用leftmost记录当前层最左结点,当前指针head作为父结点,逐层遍历为下一层的结点填充next指针。

我们一共要填充两种next指针,一种是同一个父结点的两个结点之间,一种是不同父结点的两个结点之间。第一种可直接建立,让head的左孩子的next指向右孩子;第二种需判断head是否有next,若有,则让head的右孩子指向head的next的左孩子。

head不断在本层后移,当head走到空时说明本层已遍历完,可以开始下一层,leftmost不断跳到下一层最左结点,直到走到空说明所有层都已经填充完毕。

class Solution {
public:
    Node* connect(Node* root) {
        if(!root) return nullptr;
        Node*leftmost=root;//从根结点开始
        while(leftmost->left){//当左结点不空说明还有可连接的结点
            Node*head=leftmost;// 遍历这一层结点,为下一层的节点更新 next 指针         
            while(head){
                head->left->next=head->right;//连接同一个父节点的两个子节点
                if(head->next){//在不同父亲的子节点之间建立连接
                    head->right->next=head->next->left;
                }
                head=head->next;//指针后移
            }
            leftmost=leftmost->left;//去下一层最左结点
        }
        return root;
    }
};

 思路:

本题与上一题的区别就是本题中给的二叉树不是完美二叉树,所以上一题的解法二不能用,用解法一同样可以解决。 

代码注释同上题。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值