LeetCode_114 二叉树展开为链表

题目描述:

在这里插入图片描述

思路:

由题目可以将解题过程分为以下几个步骤:
1.将当前节点的右子树连接到当前节点左子树的最右边节点的右子树上
2.将当前节点的左子树连接到右子树上,并将左子树置空
3.将当前节点的右子树作为新的当前节点,重复上述过程

查找最右侧节点未必需要写递归,写循环也可以,所以我得出做树的问题,有时候也可以尝试使用循环来解决

/**
 * 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) {
        if(root==nullptr)
            return ;
        while(root){
            if(root->left==nullptr)
                root=root->right;
            else{
                TreeNode* pre=root->left;
                //循环查找最右子节点
                while(pre->right){
                    pre=pre->right;
                }
                pre->right=root->right;
                root->right=root->left;
                root->left=nullptr;
                root=root->right;
            }
        }
    }
};
/**
 * 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:
	//递归查找最右子节点,思路其实一样
    TreeNode* findr(TreeNode* root){
        if(root->right==nullptr)
            return root;
        return findr(root->right);
    }

    void flatten(TreeNode* root) {
        if(root==nullptr)
            return ;
        while(root){
            if(root->left==nullptr)
                root=root->right;
            else{
                TreeNode* pre=findr(root->left);
                pre->right=root->right;
                root->right=root->left;
                root->left=nullptr;
                root=root->right;
            }
        }
    }
};

思路:

题目很明显是按照先序遍历去遍历的二叉树,但如果按照先序遍历,将当前节点的右子树指向左子树,则右子树会丢失,但是可以逆序思考(跟先序遍历正好相反的顺序遍历),这样右子树就不会丢失了

/**
 * 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 posttrave(TreeNode* &root,TreeNode* &pre){
        if(root==nullptr)
            return ;
        posttrave(root->right,pre);
        posttrave(root->left,pre);
        root->right=pre;
        root->left=nullptr;
        pre=root;
    }

    void flatten(TreeNode* root) {
        if(root==nullptr)
            return ;
        TreeNode* pre=nullptr;
        posttrave(root,pre);
    }
};

思路:

既然右子树会丢失,那就申请一个栈来存放树节点,这样右子树就不会丢失了

/**
 * 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) {
        if(root==nullptr)
            return ;
        stack<TreeNode*> sv;
        sv.push(root);
        TreeNode* pre=nullptr;
        while(!sv.empty()){
            TreeNode* now=sv.top();
            sv.pop();
            if(pre!=nullptr){
                pre->right=now;
                pre->left=nullptr;
            }
            if(now->right)
                sv.push(now->right);
            if(now->left)
                sv.push(now->left);
            pre=now;
        }
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
引用\[1\]中给出了一段Java代码,是关于LeetCode 114题的解法。这道题的要求是将二叉树展开为一个单链表。引用\[2\]中提到了这道题可以用递归解法,而不一定要求空间复杂度为O(1)。引用\[3\]中给出了一种解题思路,即将左子树插入到右子树的地方,然后将原来的右子树接到左子树的最右边节点,最后重复这个过程直到新的右子树为null。 根据以上引用内容,可以使用递归的方式来解决LeetCode 114题。具体的Python代码如下: ```python class Solution: def flatten(self, root: TreeNode) -> None: """ Do not return anything, modify root in-place instead. """ if not root: return self.flatten(root.left) self.flatten(root.right) if root.left: temp = root.right root.right = root.left root.left = None while root.right: root = root.right root.right = temp ``` 这段代码首先判断根节点是否为空,如果为空则直接返回。然后递归地对左子树和右子树进行展开。如果根节点的左子树不为空,将右子树保存到临时变量temp中,然后将左子树赋值给右子树,左子树置为空。接着,找到新的右子树的最右边节点,将temp连接到该节点的右侧。 这样,通过递归的方式,就可以将二叉树展开为一个单链表。 #### 引用[.reference_title] - *1* *2* *3* [【LeetCode 114二叉树展开链表(Python)](https://blog.csdn.net/Lucy_R/article/details/107827298)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insert_down1,239^v3^insert_chatgpt"}} ] [.reference_item] [ .reference_list ]
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值