Leetcode 114. 二叉树展开为链表

JehanRio

已于 2023-10-04 09:33:22 修改

阅读量85

点赞数 1

分类专栏： Leetcode专栏文章标签：链表 leetcode 数据结构

于 2023-05-20 15:16:08 首次发布

本文链接：https://blog.csdn.net/weixin_51322383/article/details/130781826

版权

Leetcode专栏专栏收录该内容

278 篇文章 2 订阅

订阅专栏

文章提供了两种C++代码实现，用于将二叉树展开为链表。方法是通过递归遍历二叉树，重新连接节点的左右子树，使得二叉树水平展开。主要函数包括`flatten`和`traversal`，分别在首刷和二刷时使用，实现了相同的功能。

摘要由CSDN通过智能技术生成

题目

在这里插入图片描述
Leetcode 114. 二叉树展开为链表

代码（首刷自解）

/**
 * 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) {
        track(root);
    }
    TreeNode* track(TreeNode* root) {
        if(!root) 
            return root;
        track(root->left);
        track(root->right);
        auto left = root->left;
        auto right = root->right;
        root->left = nullptr;
        root->right = left;
        auto p = root;
        while(p->right) {
            p = p->right;
        }
        p->right = right;
        return root;
    }
};

代码（10.4 二刷看解析）

class Solution {
public:
    void flatten(TreeNode* root) {
        traversal(root);
    }
    TreeNode* traversal(TreeNode* root) {
        if(!root)
            return root;
        traversal(root->left);
        traversal(root->right);
        auto left = root->left, right = root->right;
        root->left = nullptr;
        root->right = left;
        auto p = root;
        while(p->right)
            p = p->right;
        p->right = right;
        return root;
    }
};