leetcode hot100 之 二叉树展开为链表

题目

给定一个二叉树，将其按照先序遍历的方式展开成一个链表。left 始终为 NULL， right 为下一个节点。

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

原题链接：https://leetcode-cn.com/problems/flatten-binary-tree-to-linked-list/

思路

直接按先序遍历进行写递归即可，同时用一个全局的 pre 变量来记录遍历过程中的上一个节点。需要注意的是，因为我们会对根节点的 left、right 进行操作，所以需要额外开辟一个指针保留左右孩子，以便后续的遍历。

• 复杂度分析
  • 时间复杂度 O(n)。遍历所有节点。
  • 空间复杂度 O(n)。

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    void flatten(TreeNode* root) {
        TreeNode* pre = new TreeNode();
        preorderTravel(root, pre);
    }
    void preorderTravel(TreeNode* root, TreeNode*& pre) {
        if (root == NULL) {
            return;
        }
        pre->right = root;
        pre->left = NULL;
        pre = root;
        TreeNode* left = root->left;
        TreeNode* right = root->right;
        preorderTravel(left, pre);
        preorderTravel(right, pre);
    }
};