113. 路径总和 II
113. 路径总和 II
给你二叉树的根节点 root
和一个整数目标和 targetSum
,找出所有 从根节点到叶子节点 路径总和等于给定目标和的路径。
叶子节点 是指没有子节点的节点。
示例 1:
输入:root = [5,4,8,11,null,13,4,7,2,null,null,5,1], targetSum = 22
输出:[[5,4,11,2],[5,8,4,5]]
示例 2:
输入:root = [1,2,3], targetSum = 5
输出:[]
示例 3:
输入:root = [1,2], targetSum = 0
输出:[]
提示:
- 树中节点总数在范围
[0, 5000]
内 -1000 <= Node.val <= 1000
-1000 <= targetSum <= 1000
深度优先搜索,一个节点走到底,不行时再返回,类似回溯算法,注意每次做出选择后记得撤销选择
class Solution {
public:
vector<vector<int>>res;
vector<int>path;
vector<vector<int>> pathSum(TreeNode* root, int targetSum) {
if(root==nullptr)return res;
DFS(root,targetSum);
return res;
}
void DFS(TreeNode* root,int targetSum){
//如果节点为空 直接返回
if(root==nullptr)return;
//到达叶子节点
if(root->left==nullptr&&root->right==nullptr){
//如果到达叶子节点路径和刚好等于目标值
if((targetSum-root->val)==0){
//记得撤销选择
path.push_back(root->val);
res.push_back(path);
path.pop_back();
}
return;
}
//做出选择
path.push_back(root->val);
DFS(root->left,targetSum-root->val);
DFS(root->right,targetSum-root->val);
//撤销选择
path.pop_back();
}
};
本文来自博客园,作者:{BailanZ},转载请注明原文链接:https://www.cnblogs.com/BailanZ/p/16128827.html