力扣114. 二叉树展开为链表

  • 思路:
    • 根据二叉树前序遍历:根-左子树-右子树;

    • 要按照前序遍历将二叉树展开,则遍历节点右子树需要挂载到左子树"最右"节点右子树上;

    • 则当前节点 current 左子树 next = current->left 的最右节点 rightmost :
      *

      cpp 复制代码
      TreeNode* rightmost = next;
      while (rightmost->right != nullptr) {
          rightmost = rightmost->right;
      }
    • 将当前节点右子树挂载到左子树"最右"节点的右子树上:rightmost->right = current->right;

    • 则当前节点 current 展开完成,将其左子树按照要求置 nullptr,右子树挂载其左子树节点:current->left = nullptr;current->right = next;

    • 迭代下一个需要展开的节点对应的树;

cpp 复制代码
/**
 * 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 *current = root;
        while (current != nullptr) {
            if (current->left != nullptr) {
                TreeNode* next = current->left;
                TreeNode* rightmost = next;
                while (rightmost->right != nullptr) {
                    rightmost = rightmost->right;
                }
                rightmost->right = current->right;

                current->left = nullptr;
                current->right = next;
            }

            current = current->right;
        }
    }
};
相关推荐
2401_8812444027 分钟前
Treap树
数据结构·算法
乌萨奇也要立志学C++28 分钟前
二叉树OJ题(单值树、相同树、找子树、构建和遍历)
数据结构·算法
网安INF30 分钟前
深度学习中的逻辑回归:从原理到Python实现
人工智能·python·深度学习·算法·逻辑回归
wsxqaz42 分钟前
浏览器原生控件上传PDF导致hash值不同
算法·pdf·哈希算法
NAGNIP1 小时前
Transformer注意力机制——MHA&MQA&GQA
人工智能·算法
摘星编程1 小时前
多模态AI Agent技术栈解析:视觉-语言-决策融合的算法原理与实践
人工智能·算法·多模态ai·视觉语言融合·ai决策算法
NAGNIP1 小时前
一文搞懂KV-Cache
人工智能·算法
CoovallyAIHub1 小时前
RTMPose:重新定义多人姿态估计的“实时”标准!
深度学习·算法·计算机视觉
爱喝茶的小茶1 小时前
周赛98补题
开发语言·c++·算法
小庞在加油2 小时前
《dlib库中的聚类》算法详解:从原理到实践
c++·算法·机器学习·数据挖掘·聚类