leetcode113路径总和II

最新推荐文章于 2024-09-06 16:32:48 发布

原创最新推荐文章于 2024-09-06 16:32:48 发布 · 179 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种算法，用于解决给定二叉树及目标和的情况下，寻找所有从根节点到叶子节点路径总和等于目标值的问题。通过递归方式遍历二叉树的所有路径，并使用栈结构记录节点值，当路径总和等于目标值且达到叶子节点时，将该路径保存为结果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

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

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

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

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

思路：递归搜索每条路径，用cnt相加每个节点上的值，用一个vector数组，来模拟栈，依次将节点加入，当cnt==sum时，

将这条路径加入res中，同时，将上一条路径的后一个节点出栈,即vec.pop_back()

class Solution {
public:
    vector<vector<int>> pathSum(TreeNode* root, int sum) {
        vector<vector<int> >  res;
        vector<int> vec;
        int cnt = 0;
        generate(root,cnt,sum,vec,res);
        return res;
    }
private:
    void generate(TreeNode *node,int &cnt,int sum,
                  vector<int> &vec,
                  vector<vector<int> > &res)
{
        if(!node)
        return;
        cnt+=node->val;
        vec.push_back(node->val);
        if(!node->left&&!node->right&&cnt==sum){
            res.push_back(vec);
        }
        generate(node->left,cnt,sum,vec,res);
      
        generate(node->right,cnt,sum,vec,res);
        vec.pop_back();
        cnt-=node->val;
}
};