Leetcode_116_Populating Next Right Pointers in Each Node

本文是在学习中的总结，欢迎转载但请注明出处：http://blog.csdn.net/pistolove/article/details/43532817

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

.

Note:

You may only use constant extra space.
You may assume that it is a perfect binary tree (ie, all leaves are at the same level, and every parent has two children).

For example,
Given the following perfect binary tree,

1
/  \
2    3
/ \  / \
4  5  6  7

After calling your function, the tree should look like:

1 -> NULL
/  \
2 -> 3 -> NULL
/ \  / \
4->5->6->7 -> NULL

思路：

（1）题意为给定一颗二叉树（可以当做满二叉树），将其每层上的节点从左到右链接起来形成链表。

（2）本文主要使用队列来实现。大体思路为：从根节点开始，按从树根向下的顺序依次将每层的节点存入队列中。首先，将根节点存入队列中，只要队列不为空，就让队头元素出队，然后判断该元素是否有左右孩子，如果有分别将左右孩子加入队列中；然后，二叉树是满二叉树，所以可以求出每层的节点数目，依次为1,2,4,8,....，通过临时变量记录当前的层次level，通过Math.power(2, level)，求得当前层上的节点数levelcount，还需要使用变量count记录当前层次上遍历了多少节点，这样每当遍历的节点数和levelcount相同时，说明当前层已遍历到最后一个节点，将该节点的next置为null，继续进行下一层的遍历。由于每次出队的总是第一个元素，所以只需将出队后的元素next置为当前队列第一个元素即可；循环遍历，直到所有节点进队列、出队列为止，最后所得即为结果。

（3）该题的思路和“按层次遍历二叉树”类似，感兴趣可以参照“按层次输出二叉树”。只不过本文所示算法效率不是很高，但是比较容易理解吧。

（4）希望本文对你有所帮助。

算法代码实现如下：

/**
*
* @author liqq
*/
public static void connect(TreeLinkNode root) {
if(root ==null) return ;
if(root!=null&&root.left==null&&root.right==null) root.next=null;

List<TreeLinkNode> list = new LinkedList<TreeLinkNode>();
list.add(root);
int count = 0;
int level = 0;
int levelcount = 1;
while(list.size()!=0){
TreeLinkNode fis = list.remove(0);
count++;
if(list.size()>=0 && count!=levelcount){
TreeLinkNode sec = list.get(0);
fis.next = sec;
}else{
fis.next =null;
level++;
levelcount = (int) Math.pow(2, level);
count=0;
}

if(fis.left!=null){
list.add(fis.left);
}

if(fis.right!=null){
list.add(fis.right);
}
}
}