LeetCode 题解

积跬步，至千里。

路径总和 II

分类：LeetCode 标签： 树 深度优先搜索 发布于：2019-10-20 ID：113

题目描述

来源于 https://leetcode-cn.com/

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

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

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

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

返回:

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

解法：深度优先遍历

在遍历过程中记录下当前走过的路径，如果路径之和等于目标值，且当前节点为叶子节点，那么将当前路径插入结果中。

深度优先遍历采用递归的写法，可以使用单个容器记录下当前的路径。进入节点时，在路径中插入节点，退出节点时，删除节点。

class Solution {
public:
    vector<vector<int>> pathSum(TreeNode* root, int sum) {
        vector<vector<int>> ret;
        vector<int> path;
        
        dfs(root, path, sum, ret);
        return ret;
    }

    void dfs(TreeNode* node, vector<int>& path, int remain, vector<vector<int>>& ret){
        if(node == nullptr){
            return;
        }
        remain -= node->val;
        path.push_back(node->val);
    
        if(node->left == nullptr && node->right == nullptr && remain == 0){
            ret.push_back(path);
        }
        
        dfs(node->left, path, remain, ret);
        dfs(node->right, path, remain, ret);
        path.pop_back();
    }
};