Leetcode-每日一题【114.二叉树展开为链表】

题目

给你二叉树的根结点 root ，请你将它展开为一个单链表：

展开后的单链表应该同样使用 TreeNode ，其中 right 子指针指向链表中下一个结点，而左子指针始终为 null 。
展开后的单链表应该与二叉树 先序遍历 顺序相同。
 

示例 1：

输入：root = [1,2,5,3,4,null,6]
输出：[1,null,2,null,3,null,4,null,5,null,6]

示例 2：

输入：root = []
输出：[]

示例 3：

输入：root = [0]
输出：[0]

提示：

• 树中结点数在范围 [0, 2000] 内
• -100 <= Node.val <= 100

解题思路

首先带大家回顾一下二叉树的后序遍历

class Node {
    int val;
    Node left;
    Node right;
    Node(int val) {
        this.val = val;
    }
}

public class BinaryTree {
    /**
     * 后序遍历
     */
    public void postorderTraversal(Node root) {
        if (root != null) {
            postorderTraversal(root.left);
            postorderTraversal(root.right);
            System.out.print(root.val + " ");
        }
    }
}

首先，定义了一个 Node 类表示二叉树的节点。节点包含了一个整型的 val 值，以及左右两个子节点的引用。
然后，定义了 BinaryTree 类，该类包含了一个方法 postorderTraversal，用于实现后序遍历。
后序遍历的顺序是：先遍历左子树，然后遍历右子树，最后遍历根节点。
如果二叉树为空，则直接返回。否则，先输出左子树的值，然后递归遍历右子树和根节点。

回顾了二叉树的后序遍历后，我们来看一下这道题 

1.我们利用递归来解决这个问题，也就是二叉树的遍历思想，在还没操作节点右子树前，不能破坏该节点的右子树指向。所以采用后序遍历。

2.将root的右子树先存在 temp 节点中，然后令root.right = root.left ，不要忘记要将root.left节点置空，之后我们对root节点进行判断若root节点还存在右子树，我们就令 root = root.right ，直到找到最后一个右子树，将temp中保存的之前的右子树接在最后一个右子树的右边即可。

代码实现

class Solution {
    public void flatten(TreeNode root) {
        if(root == null){
            return;
        }
        flatten(root.left);
        flatten(root.right);
        TreeNode temp = root.right;
        root.right = root.left;
        root.left = null;
        while(root.right != null) root = root.right;
        root.right = temp;

    }
}

测试结果