LeetCode 114. 二叉树展开为链表

最新推荐文章于 2024-09-17 17:01:36 发布

Ailecccc

最新推荐文章于 2024-09-17 17:01:36 发布

阅读量305

点赞数

分类专栏： LeetCode

本文链接：https://blog.csdn.net/Ailecccc/article/details/103287798

版权

LeetCode 专栏收录该内容

109 篇文章 0 订阅

订阅专栏

LeetCode 114. 二叉树展开为链表

题目

给定一个二叉树，原地将它展开为链表。

例如，给定二叉树

/
2 5
/ \
3 4 6
将其展开为：

1

2

3

4

5

6

分析

每次将当前结点的右节点接到左节点的最右结点上，让后用左节点替换右节点。

C++代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    void flatten(TreeNode* root) {
        if(!root){
            return;
        }
        while(root->left != NULL || root->right != NULL){
            if(root->left != NULL)
                helper(root);
            root = root->right;
        }
    }
    void helper(TreeNode* root){
        TreeNode* temp = root->left;
        while(temp->right != NULL){
            temp = temp->right;
        }
        temp->right = root->right;
        root->right = root->left;
        root->left = NULL;
    }
};/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    void flatten(TreeNode* root) {
        if(!root){
            return;
        }
        while(root->left != NULL || root->right != NULL){
            if(root->left != NULL)
                helper(root);
            root = root->right;
        }
    }
    void helper(TreeNode* root){
        TreeNode* temp = root->left;
        while(temp->right != NULL){
            temp = temp->right;
        }
        temp->right = root->right;
        root->right = root->left;
        root->left = NULL;
    }
};