leetcode: Populating Next Right Pointers in Each Node II

和1的区别在于不是完全的二叉树,因而会出现左右节点的各种分布情况。   这里还是利用while来找到每个节点的next节点,由于这个过程中我们需要知道每个节点父节点的next节点,因而在遍历的过程中一定要按照先右子结点,再左子结点的顺序!!! 这样可以保证遍历过程中每个节点的父节点的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) {
        fun(root);
        return ;
    }
    void fun(TreeLinkNode root)
    {
        if(root==null)
        {
            return ;
        }
        if(root.left!=null)
        {
            if(root.right!=null)
            {
                root.left.next = root.right;
            }
            else 
            {
                TreeLinkNode tmp = root.next;
                while(tmp!=null)
                {
                    if(tmp.left!=null)
                    {
                        root.left.next = tmp.left;
                        break;
                    }
                    else if(tmp.right!=null)
                    {
                        root.left.next = tmp.right;
                        break;
                    }
                    tmp = tmp.next;
                }
            }
        }
        if(root.right!=null)
        {
            TreeLinkNode tmp = root.next;
            while(tmp!=null)
            {
                if(tmp.left!=null)
                {
                    root.right.next = tmp.left;
                    break;
                }
                else if(tmp.right!=null)
                {
                    root.right.next = tmp.right;
                    break;
                }
                tmp = tmp.next;
            }
        }
        fun(root.right);
        fun(root.left);

        
    }
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值