114. 二叉树展开为链表 Krains 2020-08-02 08:59:00 树

# 题目链接

# 前序遍历

解题思路

二叉树是按照先序遍历展开的，递归时只要记住它先序遍历的之前的一个结点pre，再把当前节点插入到pre的右子树即可
注意要存储root的左右结点信息，因为先序遍历修改了root的左右结点

class Solution {
    TreeNode pre = null;

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

        // 左子树清空
        root.left = null;
        // 如果当前root不是根结点，pre就不为null
        if(pre != null){
            pre.right = root;
        }
        // 记录先序遍历的前一个结点
        pre = root;
        
        flatten(l);
        flatten(r);
    }
}

# 后序遍历

解题思路

后续遍历返回的时候我们已经知道root的左右结点信息，这时候我们可以对其展开
对于当前root的左孩子，直接修改为root的右孩子，而对于原来root的右孩子，由于其左子树的结点不止一个，左孩子可能是左子树链表的表头，因此需要将其放在链表的末尾（用测试用例举例说明，当root为1的时，其左孩子是2，2是链表2-3-4的表头，需要将右孩子5放在4的后面）

class Solution {
    public void flatten(TreeNode root) {
        helper(root);
    }

    public TreeNode helper(TreeNode root){
        if(root == null)
            return null;
        TreeNode l = helper(root.left);
        TreeNode r = helper(root.right);
        root.right = l;
        root.left = null;

        // 将右孩子放在root链表的末尾
        TreeNode t = root;
        while(t.right != null){
            t = t.right;
        }
        t.right = r;
        
        return root;
    }
}