思路:递归 + 回溯,递归函数 path 设计如下。
参数与返回值:参数要有根节点 root,最终的字符串数组 res,存储每一条路径经过的节点的列表 path;无返回值。
终止条件:当前节点是叶子结点时,对 path数组 进行处理,得到符合题意的字符串 spath,将处理过的 spath 加入 res 中,return。
单层递归逻辑(对 spath 的处理):
- 在终止条件前执行的代码:将 root.val 加入 path。
- 在终止条件后执行的代码:当 root.left 不为空时,调用递归函数 path,再进行回溯(对 path 数组进行弹出);当 root.right 不为空时,调用递归函数 path,再进行回溯(对 path 数组进行弹出)。
class Solution(object):
def path(self,root,res,path):
path.append(root.val)
if root.left==None and root.right==None:
spath=''
for i in range(len(path)-1):
spath=spath+str(path[i])+'->'
spath=spath+str(path[len(path)-1])
res.append(spath)
return
if root.left:
self.path(root.left,res,path)
path.pop()
if root.right:
self.path(root.right,res,path)
path.pop()
def binaryTreePaths(self, root):
res=[]
if root==None:
return res
path=[]
self.path(root,res,path)
return res