0129._sum_root_to_leaf_numbers.md

129. Sum Root to Leaf Numbers

难度: Medium

刷题内容

原题连接

• https://leetcode.com/problems/sum-root-to-leaf-numbers/

内容描述

Given a binary tree containing digits from 0-9 only, each root-to-leaf path could represent a number.

An example is the root-to-leaf path 1->2->3 which represents the number 123.

Find the total sum of all root-to-leaf numbers.

Note: A leaf is a node with no children.

Example:

Input: [1,2,3]
    1
   / \
  2   3
Output: 25
Explanation:
The root-to-leaf path 1->2 represents the number 12.
The root-to-leaf path 1->3 represents the number 13.
Therefore, sum = 12 + 13 = 25.
Example 2:

Input: [4,9,0,5,1]
    4
   / \
  9   0
 / \
5   1
Output: 1026
Explanation:
The root-to-leaf path 4->9->5 represents the number 495.
The root-to-leaf path 4->9->1 represents the number 491.
The root-to-leaf path 4->0 represents the number 40.
Therefore, sum = 495 + 491 + 40 = 1026.

解题方案

思路 1 - 时间复杂度: O(2^N)- 空间复杂度: O(1)******

递归，beats 77.51%

class Solution:
    def sumNumbers(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """
        self.res = 0
        def helper(node, path):
            if node and not node.left and not node.right:
                self.res += int(''.join(str(i) for i in path+[node.val]))
            else:
                if node.left:
                    helper(node.left, path+[node.val])
                if node.right:
                    helper(node.right, path+[node.val])
        if not root:
            return 0
        helper(root, [])
        return self.res