226. Invert Binary Tree

发布时间:2019-06-12 发布网站:脚本宝典
脚本宝典收集整理的这篇文章主要介绍了226. Invert Binary Tree脚本宝典觉得挺不错的,现在分享给大家,也给大家做个参考。

题目链接Invert Binary Tree

思路
如果需要反转一个二叉树,那么我们需要遍历整个树的所有节点。
如果想遍历所有的节点,我们可以用Depth First Search(DFS)或者Breadth First Search(BFS)。
这两种办法分别可以用迭代(iterative)或者递归(recursive)的办法实现。

算法复杂度

递归:

时间:O(n) where n is the number of nodes
空间:O(n)

DFS/BFS:

时间:O(n) where n is the number of nodes
空间:O(n)

代码

递归:

class Solution(object):
    def invertTree(self, root):
        """
        :type root: TreeNode
        :rtype: TreeNode
        """
        if root:
            root.left, root.right = self.invertTree(root.right), self.invertTree(root.left)
        return root

DFS (Stack):

class Solution(object):
    def invertTree(self, root):
        """
        :type root: TreeNode
        :rtype: TreeNode
        """
        stack = [root]
        while stack:
            node = stack.pop()
            if node:
                node.left, node.right = node.right, node.left
                stack.extend([node.right, node.left])
        return root

BFS (Queue):

class Solution(object):
    def invertTree(self, root):
        """
        :type root: TreeNode
        :rtype: TreeNode
        """
        queue = [root]
        while queue:
            node = queue.pop(0)
            if node:
                node.left, node.right = node.right, node.left
                queue.extend([node.left, node.right])
        return root


脚本宝典总结

以上是脚本宝典为你收集整理的226. Invert Binary Tree全部内容,希望文章能够帮你解决226. Invert Binary Tree所遇到的问题。

如果觉得脚本宝典网站内容还不错,欢迎将脚本宝典推荐好友。

本图文内容来源于网友网络收集整理提供,作为学习参考使用,版权属于原作者。
如您有任何意见或建议可联系处理。小编QQ:384754419,请注明来意。
标签: