树——populating-next-right-pointers-in-each-node(层序遍历变型)

题目:


节点中多了一个next引用指向该节点的右边节点,若没有右边节点则next=null,且只能使用O(1)辅助空间。

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

例如:
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
思路:层序遍历二叉树,并操纵next引用赋值.
代码如下:
/**
 * Definition for binary tree with next pointer.
 * public class TreeLinkNode {
 *     int val;
 *     TreeLinkNode left, right, next;
 *     TreeLinkNode(int x) { val = x; }
 * }
 */
import java.util.*;
public class Solution {
    public void connect(TreeLinkNode root) {
        if(root == null)
            return;
        LinkedList<TreeLinkNode> list=new LinkedList();
        
        list.add(root);
        int cur=1;//list中存储的二叉树当前层节点数;
        int next=0;//list中存储的二叉树下一层节点数;
            while(list.size()!=0)
                {
                TreeLinkNode node=list.remove(0);
                cur--;
                if(cur!=0)//若node右边有节点,next指向右边节点;
                    {
                    node.next=list.get(0);
                }else//若node右边无节点,next指向null;
                    {
                    node.next=null;
                }
                
                if(node.left!=null)
                    {
                    list.add(node.left);
                    next++;
                }
                if(node.right!=null)
                    {
                    list.add(node.right);
                    next++;
                }
                
                if(cur == 0)//当前层遍历结束,遍历下一层;
                    {
                    cur=next;
                    next=0;
                }
            }
    }
}



节点中多了一个next引用指向该节点的右边节点,若没有右边节点则next=null.
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值