和 116 一个类型,但是稍微tricky一点,因为不是balance tree,而是更加 general的任意tree,也是要把next level 链接起来。
所以对应的处理方法就是要有 dummyNodeForNextLevel 作为每一层的起点,同时该层需要一个 curr 以dummyNodeForNextLevel,对该层进行延伸!
两者的分工明确,dummyNodeForNextLevel是为了从上一层走到下一层,而curr是把下一层的每个node都 hook up上。
nice summary!!
代码:
/**
* Definition for binary tree with next pointer.
* public class TreeLinkNode {
* int val;
* TreeLinkNode left, right, next;
* TreeLinkNode(int x) { val = x; }
* }
*/
public class Solution {
public void connect(TreeLinkNode root) {
while(root!=null){ // same as while(!currLevel.isEmpty()){...} // the 1st while(root!=null) check the first head of the level
TreeLinkNode nextLevelDummy= new TreeLinkNode(0); // 因为不确定下一层的head在什么位置,因此需要dummyNode
TreeLinkNode nextLevelPreScaner=nextLevelDummy;
// the 2nd while(root!=null) check whether it reach the end of the level.
while(root!=null){ // link the next level !! this is different than the ArrayList collection of node! 之前的ArrayList collection只是为了统一的收集的操作,我的思路也就只停留在这个思路。实际上对下一层的ops不是只能收集啊,可以在遍历的时候就新进拼接了!!和version I中的不一样是,那里说好了是balanced binary tree,也就是说有左节点就肯定有右节点。因此直接通过 curr.left 和 curr.right就可以链接了。但是!!这里并没有.left 和 .right的确定关系,所有需要变量来记录,最后那的tail在哪里。这是最关键的一点!所以有了前面的2个 pointer reference
if(root.left!=null){
nextLevelPreScaner.next=root.left;
nextLevelPreScaner=nextLevelPreScaner.next;
}
if(root.right!=null){
nextLevelPreScaner.next=root.right;
nextLevelPreScaner=nextLevelPreScaner.next;
}
root=root.next; // 第一层时,就到null了。同时第二层的 横向连接也全部完成了。
}
root=nextLevelDummy.next;
}
}
}
// 对比了4道题,这4道题都可以放到一起总结:levelTraversal, zigzagLevelTravesal, populatingNextPointerForBalanceBT, populatingNextPointerforBT.
// 这4道题就对 横向的层序遍历有了个很好的总结了。统一思路:while()对当前层的循环,while()内部有两件事:1,currLevel;2,nextLevel。针对这两层,可以有不同的ops,对curr,1)对val的遍历;2)对左右子节点的遍历(a,全部收集到list里,b,进行next的横向连接 II,balance tree;II,unbalanced tree)
clear version
/**
* Definition for binary tree with next pointer.
* public class TreeLinkNode {
* int val;
* TreeLinkNode left, right, next;
* TreeLinkNode(int x) { val = x; }
* }
*/
public class Solution {
public void connect(TreeLinkNode root) {
while(root!=null){ // same as while(!currLevel.isEmpty()){...} // the 1st while(root!=null) check the first head of the level
TreeLinkNode nextLevelDummy= new TreeLinkNode(0); // 因为不确定下一层的head在什么位置,因此需要dummyNode
TreeLinkNode nextLevelPreScaner=nextLevelDummy;
while(root!=null){
if(root.left!=null){
nextLevelPreScaner.next=root.left;
nextLevelPreScaner=nextLevelPreScaner.next;
}
if(root.right!=null){
nextLevelPreScaner.next=root.right;
nextLevelPreScaner=nextLevelPreScaner.next;
}
root=root.next; // 第一层时,就到null了。同时第二层的 横向连接也全部完成了。
}
root=nextLevelDummy.next;
}
}
}