剑指offer 面试题34 二叉树中和为某一值的路径

1.题目

输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。

2.思路

是否存在一个累加为目标和的路径,我们可以把目标和(sum) 减去每个路径上节点的值,来进行递归。直到它到达叶节点并且差值等于0(sum == node->val 或者说 sum - node->val == 0),这样得到了一条路径。 

3.实现

class Solution {
public:
    vector<vector<int>> FindPath(TreeNode* root,int expectNumber) {
        vector<vector<int>> results;
        vector<int> result;
        
        if(root == nullptr)
            return results;
        
        helper(root, expectNumber, result, results);
        
        return results;
    }
    
    void helper(TreeNode* root, int sum, vector<int> result, vector<vector<int>>& results)
    {
        if(root == nullptr)
            return;
        
        result.push_back(root->val);
        
        if(root->val == sum && root->left == nullptr && root->right == nullptr)
            results.push_back(result);
        
        helper(root->left, sum - root->val, result, results);
        helper(root->right, sum - root->val, result, results);
    }
};

猜你喜欢

转载自blog.csdn.net/u014128608/article/details/93488298