Sunday, February 24, 2019

LeetCode 94. Binary Tree Inorder Traversal

'''
Given a binary tree, return the inorder traversal of its nodes' values.

'''
# 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):
    #recursive
    def inorderTraversal1(self, root):
        """
        :type root: TreeNode
        :rtype: List[int]
        """
        def inorder(node):
            if node.left:
                inorder(node.left)
            self.result.append(node.val)
            if node.right:
                inorder(node.right)
       
        self.result = []
        if root:
            inorder(root)
        return self.result

    #iterative
    def inorderTraversal(self, root):
        res, stack = [], []
        while True:
            while root:
                stack.append(root)
                root = root.left
            if not stack:
                return res
            node = stack.pop()
            res.append(node.val)
            root = node.right

No comments:

Post a Comment