代码随想录-- 二叉树 -- 二叉树的所有路径
257. 二叉树的所有路径 - 力扣(LeetCode)
思路:递归 + 回溯,递归函数 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)returnif 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 respath=[]self.path(root,res,path)return res