Python实现"翻转二叉树"的方法

最新推荐文章于 2022-12-05 18:06:03 发布

求兵

最新推荐文章于 2022-12-05 18:06:03 发布

阅读量2.6k

点赞数

分类专栏： Algorithms

本文链接：https://blog.csdn.net/qiubingcsdn/article/details/82867056

版权

Algorithms 专栏收录该内容

106 篇文章 5 订阅

订阅专栏

翻转一颗二叉树

Example:

Input:

     4
   /   \
  2     7
 / \   / \
1   3 6   9

Output:

     4
   /   \
  7     2
 / \   / \
9   6 3   1

1：递归方法

def invertTree(self, root):
        """
        :type root: TreeNode
        :rtype: TreeNode
        """
        if not root:
            return None
        newRoot = TreeNode(root.val)
        newRoot.left = self.invertTree(root.right)
        newRoot.right = self.invertTree(root.left)
        return newRoot

2：迭代方法

def invertTree(self, root):
        """
        :type root: TreeNode
        :rtype: TreeNode
        """
        if not root:
            return None
        nodeList = [root]
        while nodeList:
            node = nodeList.pop(0)
            if node:
                nodeList.append(node.left)
                nodeList.append(node.right)
                node.left, node.right = node.right, node.left
        return root

算法题来自：https://leetcode-cn.com/problems/invert-binary-tree/description/