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

最新推荐文章于 2020-07-22 11:45:03 发布

abc_xian

最新推荐文章于 2020-07-22 11:45:03 发布

阅读量69

点赞数

CC 4.0 BY-SA版权

分类专栏： c++

本文链接：https://blog.youkuaiyun.com/abc_xian/article/details/99762668

c++ 专栏收录该内容

72 篇文章

订阅专栏

本文介绍了一种算法，用于查找二叉树中所有节点值之和等于给定整数的路径。通过前序遍历，将节点值累加至路径上，遇到叶节点时检查路径总和是否等于目标值，若相等则保存路径。递归遍历左右子树，回溯时删除当前节点。

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

题目描述

输入一颗二叉树的根节点和一个整数，打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中，数组长度大的数组靠前)

/*
struct TreeNode {
	int val;
	struct TreeNode *left;
	struct TreeNode *right;
	TreeNode(int x) :
			val(x), left(NULL), right(NULL) {}
};
*/
/*
当用前序遍历的方式访问到某一节点时，我们把该节点添加到路径上，并累加该节点的值(或者累减该节点的值)。
如果该节点为叶节点，并且路径中的节点值的和刚好等于输入的整数(输入的整数与路径中的节点值的差刚好等于0)
则符合当前要求，把它打印或者保存起来。
如果如果该节点不是叶节点，则继续访问它的子节点。当前节点访问结束后，递归函数自动回到它的父亲节点。
因此，在函数退出之前要在路径上删除当前节点并减去当前节点的值，以确保返回父节点时的路径刚好是从根节点到父节点。
*/
class Solution {
public:
    vector<vector<int> > buffer; //路径数组
    vector<int> tmp; //临时数组
    vector<vector<int> > FindPath(TreeNode* root,int expectNumber) 
    {
        if(root==NULL) //根节点为空
            return buffer;
        tmp.push_back(root->val); //根节点保存到临时数组
        if((expectNumber-root->val)==0 && root->left==NULL && root->right==NULL)
            //expectNumber-root->val：expectNumber的值减去root->val的值
            //如果是叶子节点且路径上节点值的和等于输入的值，保存这条路径
            {
            buffer.push_back(tmp); //存入到路径数组
            }
        FindPath(root->left,expectNumber-root->val);//如果不是叶节点，则遍历子节点
        FindPath(root->right,expectNumber-root->val);
        if(tmp.size()!=0) //返回父节点之前，路径上删除当前节点
            tmp.pop_back();
        return buffer;
    }
};