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

最新推荐文章于 2022-01-16 15:22:39 发布

原创最新推荐文章于 2022-01-16 15:22:39 发布 · 201 阅读

0 ·

CC 4.0 BY-SA版权

算法专栏收录该内容

32 篇文章

订阅专栏

本文介绍了一种算法，该算法能够找到一棵给定二叉树中所有从根节点到叶子节点的路径，这些路径上的节点值之和等于给定的整数。通过递归方法实现，详细展示了如何遍历二叉树并判断路径和。

题目描述

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

/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
public:
vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
//如果二叉树为空返回空
vector<vector<int> > ivecvec;
if (NULL == root) {
return ivecvec;
}
//二叉树不为空时，寻找根节点到叶子结点的路径，并判断和
//这是对寻找叶子结点路径算法的扩展
vector<int> ivec;//ivec用来保存路径
FindPath(ivecvec, root, ivec, 0, expectNumber);
return ivecvec;
}
//path用来保存路径, currSum表示当前路径的和
void FindPath(vector<vector<int> > &ivecvec, TreeNode *root, vector<int> path, int currSum, const int expectNumber) {
if (NULL != root) {//结点不为空，存入路径数组中
path.push_back(root -> val);
currSum += root->val;
//如果结点的左、右子树均为空，说明这个节点为叶子结点
//如果是叶子结点就要判断路径和是否等于期待值
if ((NULL == root->left) && (NULL == root->right)) {
if (currSum == expectNumber) {
ivecvec.push_back(path);
}
return;
}
//如果结点的左子树不为空
if (root->left) {
FindPath(ivecvec, root->left, path, currSum, expectNumber);
}

//如果结点的右子树不为空
if (root->right) {
FindPath(ivecvec, root->right, path, currSum, expectNumber);
}
}
return;
}

};