LeetCode116/117:填充每个节点的下一个右侧节点指针(接剑指五十六:删除链表重复节点)

题干:116

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

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

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

示例:
在这里插入图片描述

思路

对于这个题目而言,最简单的也就是递归解决,思考一下先是对根节点进行操作然后是左子树进行操作,然后若是左子树下面有结点,重复进行对应的操作,代码也比较简单:

代码

public class Solution {
    public void connect(TreeLinkNode root) {
         if(root==null || root.left==null) return; 
        root.left.next=root.right;
          if(root.next!=null){
     root.right.next=root.next.left;
      }
        connect(root.left);
        connect(root.right);
    }
}

题干:117

现上面的基本条件也都满足,只不过是,对于完美二叉树而言,可能会变成不完美二叉树,问你是否还能够像下图一样完成对应的操作。
在这里插入图片描述

思路

和上面的完美二叉树进行对比而言,对于完美二叉树时候,我们可以随意使用到

 if(root.next!=null){
     root.right.next=root.next.left;
      }
     //对于图一来说表示若2的next不为空(确实不为空,因为我们已经对2的next进行了指定)时候,进行2节点指针的右子树的next值,为2的next(即节点3的左子树)

但是现在是不完美二叉树,就需要另辟蹊径了就用到了和剑指56删除相同结点有相识之处:定义一个前驱(三界之外)结点。就利用到一个新定义的node节点,在进行这一层操作时候,总是将pre指针和当前层的左右节点定位到同一层上,在进行下一轮递归时候就能够取到当前层的结果值。

代码

具体代码如下:

public class Solution {
    public void connect(TreeLinkNode root) {
        if(root==null) 
            return;
        TreeLinkNode node=new TreeLinkNode (0);
        TreeLinkNode cur;
        TreeLinkNode pre=node;
        for(cur=root;cur!=null;cur=cur.next){
            if(cur.left!=null){
                pre.next=cur.left;
                pre=pre.next;
            }
             if(cur.right!=null){
                pre.next=cur.right;
                pre=pre.next;
            }
        }
        connect(node.next);
    }
}
发布了80 篇原创文章 · 获赞 52 · 访问量 7306

猜你喜欢

转载自blog.csdn.net/weixin_44015043/article/details/105420355