基础知识:
在Python类中规定,函数的第一个参数是实例对象本身,并且约定俗成,把其名字写为self。其作用相当于java中的this,表示当前类的对象,可以调用当前类中的属性和方法。
在类中定义的函数:第一个参数永远是 self ,并且调用时不用传递该参数;在类中函数相互调用要加 self。
题目:
给定一个二叉树,它的每个结点都存放一个 0-9 的数字,每条从根到叶子节点的路径都代表一个数字。
例如,从根到叶子节点路径 1->2->3 代表数字 123。计算从根到叶子节点生成的所有数字之和。
说明: 叶子节点是指没有子节点的节点。
思路:
利用深搜DFS,递归求解
解答:
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def sumNumbers(self, root: TreeNode) -> int:
self.res=0
def dfs(node,total):
if not node:
return
if not node.left and not node.right:
self.res+=10*total+node.val
return
total=10*total+node.val
dfs(node.left,total)
dfs(node.right,total)
dfs(root,0)
return self.res
注意:若不加self,要声明res不是局部变量
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def sumNumbers(self, root: TreeNode) -> int:
res=0
def dfs(node,total):
nonlocal res
if not node:
return
if not node.left and not node.right:
res+=10*total+node.val
return
total=10*total+node.val
dfs(node.left,total)
dfs(node.right,total)
dfs(root,0)
return res