《剑指offer》二叉树中和为某一值的路径
一、题目描述
输入一颗二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
二、输入描述
输入一颗二叉树和一个整数
三、输出描述
结点值的和为输入整数的所有路径
注意:
路径的定义是:从树的根结点开始往下一直到叶结点所经过的结点形成一条路径
四、牛客网提供的框架
/*
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) {
}
};
五、解题思路
使用先序遍历,递归找出所有的路径,把路径中各节点值得和等于expectNumber。
六、代码
#include<iostream>
#include<vector>
using namespace std;
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>> allPath;
vector<int> onePath;
vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
preOrderFind(root, expectNumber);
return allPath;
}
/*
* 先序(递归)遍历
*/
void preOrderFind(TreeNode* root, int expectNumber)
{
if(root) onePath.push_back(root->val); //一条路径,先加入根节点
if(root && root->left) preOrderFind(root->left, expectNumber); //左子树
if(root && root->right) preOrderFind(root->right, expectNumber); //右子树
int sum = 0;
for(int i = 0; i < onePath.size(); i++) //路径的节点值求和
sum += onePath[i];
if(sum == expectNumber && root && !root->left && !root->right) allPath.push_back(onePath); //该节点已经是叶子节点,判断路径的节点值得和是否等于expectNumber
onePath.pop_back();
}
};