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

谛听-

于 2018-03-06 08:39:41 发布

阅读量168

点赞数

CC 4.0 BY-SA版权

分类专栏：题树回溯

本文链接：https://blog.youkuaiyun.com/u012319493/article/details/79453235

题同时被 3 个专栏收录

456 篇文章

订阅专栏

回溯

62 篇文章

订阅专栏

树

46 篇文章

订阅专栏

本文介绍了一种算法，用于寻找给定二叉树中所有节点值之和等于特定整数的路径。通过深度优先搜索（DFS）遍历二叉树，并记录路径节点值，最终返回所有符合条件的路径。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};*/
class Solution {
public:
    void dfs(vector<vector<int> > &ans, int sum, vector<int> &path, TreeNode* root, int expectNumber)
    {
        sum += root->val;
        path.push_back(root->val);

        if(sum == expectNumber && !root->left && !root->right)
            ans.push_back(path);

        if(root->left)
            dfs(ans, sum, path, root->left, expectNumber);
        if(root->right)
            dfs(ans, sum, path, root->right, expectNumber);

        path.pop_back();
        sum -=root->val;
    }

    vector<vector<int> > FindPath(TreeNode* root, int expectNumber) 
    {
        vector<vector<int> > ans;
        if(!root)
            return ans;

        vector<int> path;
        dfs(ans, 0, path, root, expectNumber);
        return ans;
    }
};