【LeetCode刷题(中等程度)】113. 路径总和 II

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

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

示例:
给定如下二叉树,以及目标和 sum = 22,
在这里插入图片描述

返回:

[
[5,4,11,2],
[5,8,4,5]
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/path-sum-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

思路:这种求路径什么的一半都是使用深度优先搜索,但是我们要注意这里处理如何知道到当前这个节点的和是否已经为sum了,可以在每次递归进入之前减去当前节点的值,进入递归后当剩余的值正好是进入递归后的当前节点值,则满足要求了。

/**
 * 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>> res;
    vector<int> temp;
    int sum_temp = 0;
    vector<vector<int>> pathSum(TreeNode* root, int sum) {
        if(root == nullptr)
            return res;
        DFS(root,sum);
        return res;
    }

    void DFS(TreeNode*root,int sum)
    {
        if(!root)
            return;
        temp.push_back(root->val);
        if(root->val==sum&&(root->left==NULL&&root->right==NULL)){
            res.push_back(temp);
        }
        if(root->left)
        {
            DFS(root->left,sum - root->val);
        }
        if(root->right)
        {
            DFS(root->right,sum - root->val);
        }
        temp.pop_back();//退回上一层记得将弹出最后一个元素
    }
};

猜你喜欢

转载自blog.csdn.net/qq_33197518/article/details/108816522