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

最新推荐文章于 2021-09-28 17:30:07 发布

原创最新推荐文章于 2021-09-28 17:30:07 发布 · 221 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#二叉树

算法专栏收录该内容

69 篇文章

订阅专栏

本文介绍了一种算法，该算法接收一棵二叉树及一个整数作为输入，然后找出所有从根节点到叶子节点的路径，使得这些路径上的节点值之和等于给定的整数。

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

题目描述

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

代码

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};*/
class Solution {

    vector<vector<int>> array;
public:
    vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
        if (root == NULL)
            return array;
        std::vector<int> path;
        int currentSum = 0;
        array = FindPath(root,expectNumber,path,currentSum);
        return array;
    }

    vector<vector<int> > FindPath(TreeNode *pRoot, int expectNumber, std::vector<int> &path, int currentSum){
        currentSum += pRoot->val;
        path.push_back(pRoot->val);

        // 如果是叶结点，并且路径上结点和的值等于输入的值
        // 打印出这条路径
        bool isLeaf = pRoot->left == NULL && pRoot->right == NULL;
        if(isLeaf && currentSum == expectNumber){
            array.push_back(path);
        }

        // 如果不是叶结点，则遍历它的结点
        if(pRoot->left != NULL){
            FindPath(pRoot->left,expectNumber,path,currentSum);
        }
        if(pRoot->right !=NULL){
            FindPath(pRoot->right,expectNumber,path,currentSum);
        }
        path.pop_back();

        return array;

    }
};