输入一棵二叉树和一个整数,打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。
示例:
给定如下二叉树,以及目标和 sum = 22,
5
/ \
4 8
/ / \
11 13 4
/ \ / \
7 2 5 1
返回:
[
[5,4,11,2],
[5,8,4,5]
]
提示:
节点总数 <= 10000
题源链接:https://leetcode-cn.com/problems/er-cha-shu-zhong-he-wei-mou-yi-zhi-de-lu-jing-lcof
解题思路:
本问题是典型的二叉树方案搜索问题,使用回溯法解决,其包含 先序遍历 + 路径记录 两部分。
- 先序遍历: 按照 “根、左、右” 的顺序,遍历树的所有节点。
- 路径记录: 在先序遍历中,记录从根节点到当前节点的路径。当路径为 ① 根节点到叶节点形成的路径 且 ② 各节点值的和等于目标值 sum 时,将此路径加入结果列表。
算法流程:- pathSum(root, sum) 函数:
- 初始化: 结果列表 res ,路径列表 path 。
- 返回值: 返回 res 即可。
- recur(root, tar) 函数:
- 递推参数: 当前节点 root ,当前目标值 tar 。
- 终止条件: 若节点 root 为空,则直接返回。
- 递推工作:
- 路径更新: 将当前节点值 root.val 加入路径 path ;
- 目标值更新: tar = tar - root.val(即目标值 tar 从 sum 减至 00 );
- 路径记录: 当 ① root 为叶节点 且 ② 路径和等于目标值 ,则将此路径 path 加入 res 。
- 先序遍历: 递归左 / 右子节点。
- 路径恢复: 向上回溯前,需要将当前节点从路径 path 中删除,即执行 path.pop()
流程图
- pathSum(root, sum) 函数:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
vector<vector<int>> pathSum(TreeNode* root, int sum) {
// 结果二维列表
vector<vector<int>> result;
// 路径一维节点值列表
vector<int> path;
if (root == NULL){
return result;
}
// DFS-深度优先搜索,一条路走到底,然后依次回溯寻找其他路径
recur(root, sum, result, path);
return result;
}
private:
void recur(TreeNode *currentNode, int traget, vector<vector<int>> &result, vector<int> &path){
if (currentNode == NULL){
// 节点为空时,直接跳出函数
return;
}
// 将当前节点的值添加进path
path.push_back(currentNode->val);
// 目标值等于剩下目标值减去currentNode->val,目标是随着搜索的深入而减小的
traget -= currentNode->val;
if (traget==0 && currentNode->left==NULL && currentNode->right==NULL){
// 路基结束的标志,target值变为0,即路径上所有value之和为sum,
// 此节点不在有左右子树
// 将此条路径压入结果列表
result.push_back(path);
}
// 递归访问节点左右子树,DFS-深度优先搜索
recur(currentNode->left, traget, result, path); // 先依次访问左子树
recur(currentNode->right, traget, result, path); // 再一层一层从底至上访问又子树
// 向上回溯之前,先删除不符合要求的节点的值
path.pop_back();
}
};