剑指offer34 二叉树中和为某一值的路径-CSDN博客

本文链接：https://blog.csdn.net/study_deep/article/details/91347140

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

输入一棵二叉树和一个整数，打印出二叉树中结点值的和为输入整数的所有路径。

从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。

样例

给出二叉树如下所示，并给出num=22。
      5
     / \
    4   6
   /   / \
  12  13  6
 /  \    / \
9    1  5   1

输出：[[5,4,12,1],[5,6,6,5]]

思路：

递归遍历路径，到叶节点的时候判断是否符合条件。

AcWing-47 C++ code：

/**
 * 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 findPathCore(TreeNode* root, int sum, int now_sum, vector<int>& temp, vector<vector<int>>& res){
        now_sum += root->val;
        temp.push_back(root->val);
        if(!root->left && !root->right && now_sum == sum){
            res.push_back(temp);
        }
        if(root->left){
            findPathCore(root->left, sum, now_sum, temp, res);
        }
        if(root->right){
            findPathCore(root->right, sum, now_sum, temp, res);
        }
        temp.pop_back();
    }
    vector<vector<int>> findPath(TreeNode* root, int sum) {
        if(root == NULL){
            return {};
        }
        vector<vector<int>> res;
        vector<int> temp;
        findPathCore(root, sum, 0, temp, res);
        return res;
    }
};