题干: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);
}
}