LeetCode 114. 二叉树展开为链表

思路:先利用DFS的思路找到最左子节点,然后回到其父节点,把其父节点和右子节点断开,将原左子结点连上父节点的右子节点上,然后再把原右子节点连到新右子节点的右子节点上,然后再回到上一父节点做相同操作。

public static void flatten(TreeNode root) {
        if(root!=null){
            flatten(root.l);
            flatten(root.r);
            TreeNode tmp = root.r;
            root.r = root.l;
            root.l = null;
            while (root.r!=null){
                root = root.r;
            }
            root.r = tmp;
        }
    }

非递归如下:

public static void flatten(TreeNode root) {
        Stack<TreeNode> stack = new Stack<>();
        while (root!=null || !stack.isEmpty()){
            while (root!=null){
                stack.add(root);
                root = root.l;
            }
            if(!stack.isEmpty()){
                root = stack.pop();
                TreeNode tmp = root.r;
                root.r = root.l;
                root.l = null;
                while (root.r!=null){
                    root = root.r;
                }
                root.r = tmp;
                root = root.r;
            }
        }
    }

 

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值