面试题34:二叉树中和为某一值的路径(递归 C++)

因为从根节点开始寻找,前序、中序、后序遍历中只有前序遍历的根节点首先出现,所以利用前序遍历的规则去进行寻找路径,当访问到一个节点时,就将节点的值加上,如果当路径和(currentSum)等于预期的值并且为叶节点的时候,将此路径压入容器(vector)中;如果访问到叶节点的路径和(currentSum)没有达到预期值,回退到上一个节点,路径和(currentSum)减去此节点的值,一直到遍历完所有的节点-------此规则符合栈的原理,但是原本书上的的题目是需要打印(需要从根节点开始打印路径),所以用stack不合适,选用vector相对来说比较合适;再者,本身递归调用的本质就是一个压栈出栈的过程,所以运用递归进行解题

/**
 * 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>> pathSum(TreeNode* root, int sum) {
        vector<vector<int>>answer;
        if(root==nullptr)return answer;
        vector<int>path;
        _pathSum(root,sum,0,answer,path);
        return answer;
    }
    void _pathSum(TreeNode*root,int sum,int currentSum,vector<vector<int>>&answer,vector<int>&path)
    {
        currentSum+=root->val;
        path.push_back(root->val);
        bool isLeaf=root->left==nullptr&&root->right==nullptr;
        if(isLeaf&&currentSum==sum)
        {//如果当前的路径和等于预期值并且处于叶节点,将路径压入容器中
            answer.push_back(path);
        }
        if(root->left)
        {//递归左子树
            _pathSum(root->left,sum,currentSum,answer,path);
        }
        if(root->right)
        {//递归右子树
            _pathSum(root->right,sum,currentSum,answer,path);
        }
        //回退,弹出节点值
        path.pop_back();
    }
};
发布了107 篇原创文章 · 获赞 28 · 访问量 1957

猜你喜欢

转载自blog.csdn.net/qq_38994205/article/details/104507109