LeetCode 113. Path Sum II

元气小坏坏 提交于 2020-01-24 01:45:41

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

  1. 从根节点深度遍历二叉树,先序遍历时,将该节点值存储至 pathpath栈 中 (vector实现),使用 path_valuepath\_value 累加节点值。
  2. 当遍历至叶结点时,检查 path_valuepath\_value 值是否为 sumsum,若为 sumsum,则将pathpath pushpush进入 resultresult 结果中。
  3. 在回溯时,将该节点值从 pathpath栈 中弹出,path_valuepath\_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;
    }
};
标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!