leetcode 113.路径总和II

leetcode 113.路径总和II

题目描述

给定一个二叉树和一个目标和，找到所有从根节点到叶子节点路径总和等于给定目标和的路径。

说明: 叶子节点是指没有子节点的节点。

示例:
给定如下二叉树，以及目标和 sum = 22

              5
             / \
            4   8
           /   / \
          11  13  4
         /  \    / \
        7    2  5   1

返回:

[
   [5,4,11,2],
   [5,8,4,5]
]

解题思路

回溯法解题

该题是 leetcode 112的升级版，点这里
先写出了回溯法解题框架

回溯法框架：递归出口设计、目标条件的选择、递归、状态重置。
对应代码如下

/**
 * 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>> res;        // 用于存储目标结果
        vector<int> track;              // 存储符合条件的路径
        backTrack(root, res, track, sum);   // 递归
        return res;
    }
    // 递归函数
    void backTrack(TreeNode* root, vector<vector<int>>& res, vector<int> track, int sum){
        //步骤一：设计递归出口
        if(!root){
            return ;
        }
        if(!root->left && !root->right){
            sum -= root->val;
            track.push_back(root->val);
            if(sum == 0){
                res.push_back(track);
            }
            track.pop_back();
            sum += root->val;
            return;
        }
        // 步骤二：目标路径查询
        sum -= root->val;
        track.push_back(root->val);
        // 步骤三：对当前节点的左右子树继续递归
        backTrack(root->left, res, track, sum);
        backTrack(root->right, res, track, sum);
        // 步骤四：状态重置
        track.pop_back();
        sum += root->val;
    }
};

欢迎大家关注我的个人公众号，同样的也是和该博客账号一样，专注分享技术问题，我们一起学习进步