二叉树中和为某一值的路径

题目

  输入一棵二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。例如输入下图中二叉树和整数22,则打印出两条路径,第一条路径包含结点10、12,第二条路径包含结点10、5和7。

思路

  1. 当用前序遍历的方式访问到某一结点时,我们把该结点添加到路径上,并累加该结点的值。
  2. 如果该结点为叶结点并且路径中结点值的和刚好等于输入的整数,则当前的路径符合要求,我们把它打印出来。如果当前结点不是叶结点,则继续访问它的子结点。
  3. 当前结点访问结束后,递归函数将自动回到它的父结点。这里要注意的是:在函数退出之前要在路径上删除当前结点并减去当前结点的值,以确保返回父结点时路径刚好是从根结点到父结点的路径。
/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};*/
class Solution {
public:
    vector<vector<int> > FindPath(TreeNode* root,int num) {
        if(root==nullptr)
            return {};
        
        vector<vector<int>> res;
        vector<int> tmp;
        FindPathCore(root,num,tmp,res);
        return res;
    }
private:
    void FindPathCore(TreeNode *node,int num,vector<int> &tmp,vector<vector<int>> &res)
    {
        tmp.push_back(node->val);
        if(node->left==nullptr&&node->right==nullptr&&num==node->val)
        {
            res.push_back(tmp);
            //tmp.pop_back();
           // return ;
        }
        if(node->left)
            FindPathCore(node->left,num-node->val,tmp,res);
        if(node->right)
            FindPathCore(node->right,num-node->val,tmp,res);
        
        tmp.pop_back();
        return ;
    }
};
原文地址:https://www.cnblogs.com/tianzeng/p/10200598.html