您的位置:首页 > 职场人生

【剑指Offer】面试题25:二叉树中和为某一值的路径

2017-07-20 16:10 232 查看
整理自剑指Offer

一:题目描述

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

二:解题思路

如下图所示的例子,给定了二叉树与整数22,则可以打印两条路径。第一条10,5,7、第二条10,12



题目中定义路径:从树的根节点开始一直到叶子结点所经过的结点形成一条路径

也就是说每条满足条件的路径都是以根节点开始,叶子结点结束,如果想得到所有根节点到叶子结点的路径(不一一定满足和为某整数的条件),需要遍历整棵树,还要先遍历根节点,所以采用先序遍历

以上面的树模拟先序遍历的过程

10-->5-->4  已近到达叶子结点,不满足要求22,因此该路径访问结束,需要访问下一个路径

因为在访问节点的过程中,我们并不知道该路径是否满足要求,所以我们每访问一个节点就要记录该结点

访问下有一个结点前,要先从结点4退回到结点5,再访问下一个结点7,因为4不在去往7的路径上,所以要在路径中将4删除

10-->5-->7  满足要求,保存该路径

访问下一个结点,从结点7回到结点5再回到结点10

10-->12,满足要求,保存该路径

整个流程如下图所示:



三:代码实现

/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
public:
void findAllPath
(TreeNode* root,int expectNumber,int currentSum,vector<vector<int> >& allpath,vector<int>& path){

//记录当前访问的结点
path.push_back(root->val);
//记录访问到当前节点 所有结点的和
currentSum+=root->val;

//判断是否为叶子结点 true-是
bool isLeaf=root->left==NULL && root->right==NULL;

//如果为叶子结点, 判断路径的和是否满足要求,满足,保存该路径
if(currentSum==expectNumber && isLeaf)
allpath.push_back(path);

if(root->left!=NULL)
findAllPath(root->left,expectNumber,currentSum,allpath,path);
if(root->right!=NULL)
findAllPath(root->right,expectNumber,currentSum,allpath,path);

//如果是叶子结点又不满足要求,退回到父结点,删除当前节点
path.pop_back();

}

vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {

vector<vector<int> > allpath;
vector<int> path;
//边界条件
if(root==NULL)
return allpath;
int currentSum=0;
findAllPath(root,expectNumber,currentSum,allpath,path);
return allpath;
}
};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  剑指Offer 二叉树