Given a binary tree
struct TreeLinkNode {
TreeLinkNode *left;
TreeLinkNode *right;
TreeLinkNode *next;
}
Populate each next pointer to point to its next right node. If there is no next right node, the next pointer should be set to NULL.
Initially, all next pointers are set to NULL.
题意
把每个树节点都指向它的同层的右边节点
思路
使用两个指针,第一个指针用来层与层之间的操作,第二个指针用来某一层中的next指向的操作。
简单来说,一个根节点的左孩子要指向它的右孩子,而一个根节点的右孩子要指向和这个根节点同层的右边节点的左孩子。
实现
public class Solution {
public void connect(TreeLinkNode root) {
if(root==null)
return;
TreeLinkNode pre=root;
TreeLinkNode cur=null;
while(pre.left!=null)
{
cur=pre;
while(cur!=null)
{
cur.left.next=cur.right;
if(cur.next!=null)
cur.right.next=cur.next.left;
cur=cur.next;
}
pre=pre.left;
}
}
}