[Python] Binary Tree Inorder Traversal

Medium 난이도 Acceptence 66.5% 문제

 

이진트리를 inorder 순회한 값을 리스트로 반환하는 문제

아래와 같이 left, root right 노드를 재귀로 순회하도록 구성하였고 왼쪽 아래의 노드부터 inorder로 순회하도록 한다.

여기서 extend라는 함수에 대해 찾아봤는데 재귀함수로 값을 더할 경우 extend를 사용, append사용시 원하는대로 값이 더해지지 않는 경우가 많다.

 

class Solution(object):
    def traverseNode(self, curNode):
        result = []
        if curNode is None:
            return result
        if curNode.left is not None:
            result.extend(self.traverseNode(curNode.left))
        if curNode is not None:
            result.extend([curNode.val])
        if curNode.right is not None:
            result.extend(self.traverseNode(curNode.right))
        return result

    def inorderTraversal(self, root):
        """
        :type root: TreeNode
        :rtype: List[int]
        """
        return self.traverseNode(root)