题意:
给定一个完美二叉树,每个树的结点除了左右子节点以外,还有一个next结点,要求设置为指向它水平右边相邻的那个结点,如果没有,那么保持为空。
分析:
题目说,不要用额外空间,那就是直接双重循环层次遍历了,其实也没啥意思,还不如复习下队列实现。
/**
* 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) {
Queue<TreeLinkNode> q = new LinkedList<>();
if(root == null)
return;
q.offer(root);
while(!q.isEmpty()){
int size = q.size();
for(int i=0; i<size; i++){
TreeLinkNode node = q.poll();
if(i != size-1)
node.next = q.peek();
if(node.left != null){
q.offer(node.left);
}
if(node.right != null){
q.offer(node.right);
}
}
}
}
}