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

Posted 2021-05-29 天津 唐秙

题目要求
  输入一棵二叉树和一个整数，打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。

思路
  先序遍历+回溯法，定义一个dfs函数，我们用来判断是否满足目标值，并且将满足的path传到ret中，dfs中，如果结点为空，我们直接跳过，每到一个结点，我们把目标值就要减掉当前结点的值，如果值为0，并且为叶子结点，那么这条路径就是满足条件的，否则就接着遍历它的子节点。最后再回溯到当前结点的位置。

图解

代码实现

class Solution {
public:
    vector<vector<int>> pathSum(TreeNode* root, int target) {
        dfs(root, target);
        return ret;
    }

    void dfs(TreeNode* node, int target)
    {
        if(node != nullptr)
        {
            target -= node->val;
            path.push_back(node->val);
            if(target == 0 && node->left == nullptr && node->right == nullptr)
            {
                ret.push_back(path);
            }
            else
            {
                dfs(node->left, target);
                dfs(node->right, target);
            }
            path.pop_back();
        }
    }
private:
    vector<int> path;
    vector<vector<int>> ret;
};