leetcode_116:Populating Next Right Pointers in Each Node

这个题目其实算是比较简单的,第一个想法就是层次遍历。使用两个队列,一个存当前层的节点,一个存下一层的节点。这样交替的遍历使用,每个队列都会存一层的所有节点,然后出队列的同时改变next引用就行了。当然空间复杂度会高一点。但肯定还有更简单的方法,也就是递归处理。得益于大前提,这是一个完全二叉树,所以肯定所有的叶子节点都会在一层,这样就好做多了。

当然我们要明确一下更改next引用时有两种情况:

  1. 当前节点是父节点的左子节点。那此时只需要得到父节点的右子节点就好了。所以需要提供父节点这个参数从而获得右节点。
  2. 当前节点是父节点的右子节点,那此时需要父节点的相邻节点的左子节点,然后进行更改。

明确了这两种情况,那么问题就好解决多了,直接看代码清楚一点:

    public Node connect(Node root) {
        if(root == null)
            return null;
        root.next = null;	//先把根节点处理一下
        helper(root, root.left);
        helper(root, root.right);
        return root;
    }
    
    private void helper(Node parent, Node root) {
        if (root == null)		//这时肯定是parent是个叶子节点,则直接返回
            return;
        if (parent.left == root) {   //当前节点是个左节点,则它的next指向相邻右节点
            root.next = parent.right;	
            helper(root, root.left);
            helper(root, root.right);
        } else {
            if (parent.next != null) {	//当前节点是个右节点,则它的next指向父节点相邻节点的左子节点
                root.next = parent.next.left;
            }
            helper(root, root.left);
            helper(root, root.right);
        }
    }

2019.12.3更新

每次重新做都有不同的发现,由于这个题目的特殊性(完全二叉树),所以根本不需要记住父节点,只需要记住一个根节点,然后直接对这个根节点的左右子树进行更改就行了,当遇到根节点是叶子节点的时候可以直接返回,因为在它的父节点已经完成了对叶子节点的next指针更新。下面的代码是更新后的,更加容易懂:

    public Node connect(Node root) {
        if (root == null)
            return root;
        helper(root);
        return root;
    }

    private void helper(Node root) {
    	//遇到叶子节点就可以直接返回,因为在它们的父节点已经完成了对叶子节点next指针的更新
        if (root.left == null && root.right == null)	
            return;
        //先更改左子树next,直接把左子树的next指向右子树
        root.left.next = root.right;
        //再更改右子树next,需判断此根节点右边还有没有节点,有则是它右边节点(next)的左子树,没有则是null
        if (root.next == null)
            root.right.next = null;
        else {
            root.right.next = root.next.left;
        }
        //递归对左、右子树进行这样的更改
        helper(root.left);
        helper(root.right);
    }
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值