113. 路径总和 II

113. 路径总和 II

给你二叉树的根节点 root 和一个整数目标和 targetSum ,找出所有 从根节点到叶子节点 路径总和等于给定目标和的路径。

叶子节点 是指没有子节点的节点。

示例 1:

img

输入: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:

img

输入: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();
    }
};
posted @ 2022-04-11 10:19  BailanZ  阅读(26)  评论(0编辑  收藏  举报