C/C++教程

leetcode114.二叉树展开为链表

本文主要是介绍leetcode114.二叉树展开为链表,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

leetcode 114.二叉树展开为链表

给你二叉树的根结点 root ,请你将它展开为一个单链表:

  • 展开后的单链表应该同样使用 TreeNode ,其中 right 子指针指向链表中下一个结点,而左子指针始终为 null 。
  • 展开后的单链表应该与二叉树 先序遍历 顺序相同。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/flatten-binary-tree-to-linked-list

解题思路

题目解读:就是将左子树和右子树展开为链表之后,先将左子树放到根节点的右边再将右子树放入原左子树的右边

这样的话就是用递归来处理

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
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)

        # 将root右子树暂存
        temp = root.right
        # 左子树放到右边
        root.right = root.left
        # 将左子树置空
        root.left = None
        # 找到原左子树的最右边
        while root.right:
            root = root.right
        root.right = temp
这篇关于leetcode114.二叉树展开为链表的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!