Leetcode 113. 路径总和 II C++

Leetcode 113. 路径总和 II

题目

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

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

测试样例

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

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

返回:

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

题解

dfs回溯
记录路径中的节点以及路径和，到叶结点时进行判断

代码

vector<vector<int>> ans;
    void search(TreeNode* root,int sum,vector<int>& num){
        num.push_back(root->val);
        if(root->right==NULL && root->left==NULL){
            if(sum == root->val) ans.push_back(num);   
            num.pop_back();
            return;
        }    
        if(root->left)  search(root->left,sum-root->val,num);
        if(root->right) search(root->right,sum-root->val,num);
        num.pop_back(); 
    }
    vector<vector<int>> pathSum(TreeNode* root, int sum) {
        if(root == NULL)    return ans;
        vector<int> num;
        search(root,sum,num);
        return ans;
    }

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/path-sum-ii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。