美文网首页
94. Binary Tree Inorder Traversa

94. Binary Tree Inorder Traversa

作者: April63 | 来源:发表于2018-05-16 22:10 被阅读0次

递归的方法,代码如下:

class Solution(object):
    def inorderTraversal(self, root):
        """
        :type root: TreeNode
        :rtype: List[int]
        """
        if root ==  None:
            return []
        if root.left == None and root.right == None:
            return [root.val]
        res = []
        self.dg(root, res)
        return res
    def dg(self, root, res):
        if root.left != None:
            self.dg(root.left, res)
        res.append(root.val)
        if root.right != None:
            self.dg(root.right,res)

我们使用一个栈来解决问题。步骤如下:

一,我们将根节点1入栈,如果有左孩子,依次入栈,那么入栈顺序为:1,2,4。由于4的左子树为空,停止入栈,此时栈为{1,2,4}。

二,此时将4出栈,并遍历4,由于4也没有右孩子,那么根据中序遍历的规则,我们显然应该继续遍历4的父亲2,情况是这样。所以我们继续将2出栈并遍历2,2存在右孩子,将5入栈,此时栈为{1,5}。

三,5没有孩子,则将5出栈并遍历5,这也符合中序遍历的规则。此时栈为{1}。

四,1有右孩子,则将1出栈并遍历1,然后将右孩子3入栈,并继续以上三个步骤即可。

栈的变化过程:{1}->{1,2}->{1,2,4}->{1,2}->{1}->{1,5}->{1}->{}->{3}->{3,6}->{3}->{}->{7}->{}。

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
    def inorderTraversal(self, root):
        """
        :type root: TreeNode
        :rtype: List[int]
        """
        if root == None:
            return []
        if root.left == None and root.right == None:
            return [root.val]
        res = []
        stack = []
        while stack or root:
            if root:
                stack.append(root)
                root = root.left
            else:
                root = stack.pop()
                res.append(root.val)
                root = root.right
        return res

相关文章

网友评论

      本文标题:94. Binary Tree Inorder Traversa

      本文链接:https://www.haomeiwen.com/subject/gugmdftx.html