LeetCode 113. Path Sum II(medium)

题目链接:点击这里
在这里插入图片描述
思路:二叉树的先序遍历

  1. 从根节点深度遍历二叉树,先序遍历时,将该节点值存储至 p a t h path 栈 中 (vector实现),使用 p a t h _ v a l u e path\_value 累加节点值。
  2. 当遍历至叶结点时,检查 p a t h _ v a l u e path\_value 值是否为 s u m sum ,若为 s u m sum ,则将 p a t h path p u s h push 进入 r e s u l t result 结果中。
  3. 在回溯时,将该节点值从 p a t h path 栈 中弹出, p a t h _ v a l u e path\_value 减去节点值。
/**
 * 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:
    void preorder(TreeNode* r, int &path_value, int sum, vector<int> &path, vector<vector<int>> &result)
    {
        if(!r)  return;
        
        path_value += r->val;
        path.push_back(r->val);
        
        if(r->left==NULL&&r->right==NULL&&path_value==sum)
            result.push_back(path);
        
        preorder(r->left, path_value, sum, path, result);
        preorder(r->right, path_value, sum, path, result);
        
        path_value -= r->val;
        path.pop_back();
    }
    
    vector<vector<int>> pathSum(TreeNode* root, int sum) {
        vector<vector<int>> result; //存储所有满足条件的路径
        vector<int> path;   //路径栈
        int path_value = 0;
        preorder(root, path_value, sum, path, result);
        return result;
    }
};
发布了694 篇原创文章 · 获赞 104 · 访问量 11万+

猜你喜欢

转载自blog.csdn.net/qq_42815188/article/details/104076431