【leetcode热题】二叉树展开为链表

题目描述

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

例如,给定二叉树

复制代码
    1
   / \
  2   5
 / \   \
3   4   6

将其展开为:

复制代码
1
 \
  2
   \
    3
     \
      4
       \
        5
         \
          6

解法:

先序遍历,始终记录下前一个节点,然后将当前节点接在前一个节点的右孩子上。

为了避免初始阶段 last 为空(即开始的时候前一个节点是不存在的),可以随意给 last 初始化一个节点,这并不影响结果。

python 复制代码
class Solution:
    def flatten(self, root: TreeNode) -> None:
        """
        Do not return anything, modify root in-place instead.
        """
        
        stack = [root]
        last = TreeNode(0)
        
        while stack:
            node = stack.pop()
            if not node:
                continue
        
            last.right = node
            last = node

            stack.append(node.right)
            stack.append(node.left)
            node.left = None
相关推荐
希望20171 天前
图论基础知识
算法·图论
m0_713541841 天前
systemverilog如何解决不能使用变量索引来进行位选择的范围指定
算法·systemverilog
七牛云行业应用1 天前
深度解析强化学习(RL):原理、算法与金融应用
人工智能·算法·金融
和编程干到底1 天前
数据结构 栈和队列、树
数据结构·算法
纪元A梦1 天前
贪心算法在GNN邻域采样问题中的深度解析
算法·贪心算法
宇钶宇夕1 天前
西门子 S7-200 SMART PLC 核心指令详解:从移位、上升沿和比较指令到流水灯控制程序实战
运维·算法·自动化
爱编程的化学家1 天前
代码随想录算法训练营第十一天--二叉树2 || 226.翻转二叉树 / 101.对称二叉树 / 104.二叉树的最大深度 / 111.二叉树的最小深度
数据结构·c++·算法·leetcode·二叉树·代码随想录
tqs_123451 天前
redis zset 处理大规模数据分页
java·算法·哈希算法
吃着火锅x唱着歌1 天前
LeetCode 1446.连续字符
算法·leetcode·职场和发展
愚润求学1 天前
【贪心算法】day10
c++·算法·leetcode·贪心算法