117 Populating Next Right Pointers in Each Node II 每个节点的右向指针 II

这是“每个节点的右向指针”问题的进阶。
如果给定的树可以是任何二叉树,该怎么办?你以前的解决方案仍然有效吗?
注意:
    你只能使用恒定的空间。
例如,
给定以下二叉树,
         1
       /  \
      2    3
     / \    \
    4   5    7
调用你的函数后,树应该看起来像这样:
         1 -> NULL
       /  \
      2 -> 3 -> NULL
     / \    \
    4-> 5 -> 7 -> NULL

详见:https://leetcode.com/problems/populating-next-right-pointers-in-each-node-ii/description/

Java实现:

/**
 * 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) {
        if(root==null){
            return;
        }
        LinkedList<TreeLinkNode> que=new LinkedList<TreeLinkNode>();
        que.offer(root);
        while(!que.isEmpty()){
            //记录本层节点的个数
            int size=que.size();
            for(int i=0;i<size;++i){
                TreeLinkNode cur=que.poll();
                //最后一个节点的next是null,不做处理
                if(i<size-1){
                    TreeLinkNode next=que.peek();
                    cur.next=next;
                }
                if(cur.left!=null){
                    que.offer(cur.left);
                }
                if(cur.right!=null){
                    que.offer(cur.right);
                }
            }
        }
    }
}

参考:https://segmentfault.com/a/1190000003465911

posted on 2018-04-05 11:49  lina2014  阅读(174)  评论(0编辑  收藏  举报

导航