☆打卡算法☆LeetCode 114、二叉树展开为链表算法解析

大家好，我是小魔龙，Unity3D软件工程师，VR、AR，虚拟仿真方向，不定时更新软件开发技巧，生活感悟，觉得有用记得一键三连哦。

一、题目

1、算法题目

“给定二叉树的根节点，将它展开为一个单链表。”

题目链接：

来源：力扣（LeetCode）

链接： 114. 二叉树展开为链表

2、题目描述

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

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

代码语言：javascript复制

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

代码语言：javascript复制

示例 2：
输入： root = []
输出： []

二、解题

1、思路分析

这道题的题意是要求将二叉树展开为单链表，单链表中的节点顺序即为二叉树的前序遍历访问各节点的顺序。

所以，可以对二叉树进行前序遍历，获得各节点访问的顺序。

但是将二叉树展开为单链表会破坏二叉树的结构，因此对二叉树的前序遍历结束之后要更新每个节点的左右子节点的信息。

实现前序遍历可以通过递归的方式。

2、代码实现

代码参考：

代码语言：javascript复制

class Solution {
    public void flatten(TreeNode root) {
        List<TreeNode> list = new ArrayList<TreeNode>();
        preorderTraversal(root, list);
        int size = list.size();
        for (int i = 1; i < size; i  ) {
            TreeNode prev = list.get(i - 1), curr = list.get(i);
            prev.left = null;
            prev.right = curr;
        }
    }

    public void preorderTraversal(TreeNode root, List<TreeNode> list) {
        if (root != null) {
            list.add(root);
            preorderTraversal(root.left, list);
            preorderTraversal(root.right, list);
        }
    }
}