二叉树:扁平化

    给定一棵二叉树,把它转化为一棵扁平的二叉树。如下面左边的二叉树,扁平化后的结果为右边的二叉树。



    方法一:利用递归的方法,按照左中右的访问顺序,把所有节点的右子树递归的链接到最左边的节点上,形成一棵向左的扁平二叉树,然后再把它翻转到右边。

    public void flatten(TreeNode root) {
        if(root == null)
            return;
        TreeNode[] tail = new TreeNode[1];    //使用数组,使递归函数中修改的值能在其父节点中生效
        tail[0] = root;
        recursion(root, tail);
        TreeNode curr = root;
        while(curr.left != null){    //从左边翻转到右边
            curr.right = curr.left;
            curr.left = null;
            curr = curr.right;
        }
    }
   public void recursion(TreeNode root, TreeNode[] tail){
        if(root == null){
            return;
        }
        if(root.left != null){    //左节点,只需记录最末端节点,什么都不做
            tail[0] = root.left;    
            recursion(root.left, tail);
        }
        if(root.right != null){    //右节点,把它链接到最左端的tail的左节点
            tail[0].left = root.right;
            root.right = null;
            tail[0] = tail[0].left;
            recursion(tail[0], tail);
        }
    }

    方法二:按照右左中的顺序递归访问,并记录访问的前一个节点,把左子树链接到右节点上

private TreeNode prev = null;

public void flatten(TreeNode root) {
    if (root == null)
        return;
    flatten(root.right);
    flatten(root.left);
    root.right = prev;
    root.left = null;
    prev = root;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值