力扣 114. 二叉树展开为链表

题目

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/flatten-binary-tree-to-linked-list
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

示例

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

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/flatten-binary-tree-to-linked-list
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

方法1：递归

如何把 root 的左右子树拉平？
按照 flatten 函数的定义，对 root 的左右子树递归调用 flatten 函数即可。

Python实现

class Solution:
    def flatten(self, root: TreeNode) -> None:
        """
        Do not return anything, modify root in-place instead.
        """
        if not root: return

        #拉长
        self.flatten(root.left)
        self.flatten(root.right)

        #左右2个分支
        left=root.left
        right=root.right

        #将左分支接在根上右边
        root.right=left
        root.left=None

        #将右分支接在左分支下
        p=root
        while(p.right != None):
            p=p.right
        p.right=right

Java实现

class Solution {
    public void flatten(TreeNode root) {
        if (root == null) {
            return;
        }
        // 递归将子结点拉成链表
        flatten(root.left);
        flatten(root.right);

        // 将传入结点与子结点拉成链表
        TreeNode left = root.left;
        TreeNode right = root.right;

        root.left = null;
        root.right = left;

        // 将原先的右子树接到当前右子树的末端
        TreeNode p = root;
        while (p.right != null) {
            p = p.right;
        }
        p.right = right;
    }
}

Java实现2

class Solution {
    public void flatten(TreeNode root) {
        if (root == null) return;

        flatten(root.left);
        flatten(root.right);

        TreeNode tmp = root.right;
        root.right = root.left;
        root.left = null;

        while (root.right != null) root = root.right;
        root.right = tmp;
    }
}