LeetCode 117 Populating Next Right Pointers in Each Node II

题目

Given a binary tree

struct TreeLinkNode {
TreeLinkNode *left;
TreeLinkNode *right;
TreeLinkNode *next;
}

Follow up for problem "Populating Next Right Pointers in Each Node".

What if the given tree could be any binary tree? Would your previous solution still work?

承接上一题，http://blog.csdn.net/xift810/article/details/20537657 把二叉树的next补完，但是不是完美二叉树了。

思路

1 关键点是旁边的概念不能用root.next.left来得出了。

2 旁边现在的意思是，右边节点中，出现第一个孩子节点的情况。

3 我一开始的代码分了很多类，算是硬写的；基本就等于我脑中的思路。AC了。

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

}
}
}
connect(root.right);
connect(root.left);
}
}

4 但是，这样很不好，后续如果回头看，根本看不懂写的是什么。

5 所以总结了下，重写。预先找到旁边孩子的结点，之后再分情况讨论，这样子，会好很多。突然想到了乘法合并率，嘿嘿。

代码

public class Solution {
public void connect(TreeLinkNode root) {
if(root==null){
return ;
}
TreeLinkNode rnext = root.next;
TreeLinkNode record = null;
while(rnext!=null){
if(rnext.left!=null){
record = rnext.left;
break;
}
if(rnext.right!=null){
record = rnext.right;
break;
}
rnext= rnext.next;
}
if(root.right!=null){
root.right.next=record;
}
if(root.left!=null){
if(root.right != null){
root.left.next = root.right;
}else{
root.left.next = record;
}
}
connect(root.right);
connect(root.left);
}
}