114 Flatten Binary Tree to Linked List

最新推荐文章于 2024-01-17 15:25:50 发布

原创最新推荐文章于 2024-01-17 15:25:50 发布 · 102 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 同时被 3 个专栏收录

200 篇文章

订阅专栏

Medium

178 篇文章

订阅专栏

Solved

160 篇文章

订阅专栏

本文介绍了一种将二叉树结构展平为链表的方法，通过两种不同的实现方式，一种使用队列进行中序遍历保存节点，另一种采用递归方式调整节点连接，实现了原地转换而无需额外的数据结构。

1 题目

Given a binary tree, flatten it to a linked list in-place.

For example, given the following tree:

The flattened tree should look like:

2 尝试解

/**
 * 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 inorderSave(TreeNode* root,queue<TreeNode*>&saver){
        if(root == NULL) return;
        saver.push(root);
        inorderSave(root->left,saver);
        inorderSave(root->right,saver);
    }
    void flatten(TreeNode* root) {
        queue<TreeNode*> saver;
        inorderSave(root,saver);
        while(!saver.empty()){
            TreeNode* temp = saver.front();
            saver.pop();
            temp->right = saver.empty()?NULL:saver.front();
            temp->left = NULL;
        }
    }
};

3 标准解

class Solution {
public:
    TreeNode* prev = NULL;
    void flatten(TreeNode* root) {
        if (root == NULL)
        return;
        flatten(root->right);
        flatten(root->left);
        root->right = prev;
        root->left = NULL;
        prev = root;
    }
};