题目
二叉树中和为某一值的路径
描述
输入一棵二叉树和一个整数,打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。
思路
本问题是典型的二叉树方案搜索问题,使用回溯法解决,其包含 先序遍历 + 路径记录 两部分
- 先序遍历: 按照 “根、左、右” 的顺序,遍历树的所有节点。
- 路径记录: 在先序遍历中,记录从根节点到当前节点的路径。当路径为 ① 根节点到叶节点形成的路径 且 ② 各节点值的和等于目标值 sum时,将此路径加入结果列表。
算法流程:
pathSum(root, sum) 函数:
- 初始化: 结果列表 pathes ,路径列表 path 。
- 返回值: 返回 pathes 即可。
recur(root,sum,pathes,[]) 函数:
- 递推参数: 当前节点 root ,当前目标值 sum 。
- 终止条件: 若节点 root 为空,则直接返回。
递推工作:
- 路径更新: 将当前节点值 root.val 加入路径 path ;
- 目标值更新: sum= sum- root.val(即目标值 从 sum 减至 0 );
- 路径记录: 当 ① root 为叶节点 且 ② 路径和等于目标值 ,则将此路径 path 加入 pathes 。
- 先序遍历: 递归左 / 右子节点。
代码实现
// 先序遍历
var pathSum = function(root, sum) {
if(!root){//当为空节点时返回空数组
return [];
}
const pathes =[];//结果数组
recur(root,sum,pathes,[])
return pathes;
};
function recur(root,sum,pathes,path){
if(!root){//当为空节点时不执行操作
return;
}
path = [...path,root.val];//深拷贝,记录遍历路径
if(!root.left&&!root.right&&sum===root.val){//当达到叶子节点时且路径的结果等于特定值时
pathes.push(path);//将此路径加入到结果队列中
return;
}
recur(root.left,sum-root.val,pathes,path);//递归遍历左子树
recur(root.right,sum-root.val,pathes,path);//递归遍历右子树
}
缺点
空间和时间效率不是太高,后面考虑回溯算法实现。