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

面试题25:二叉树中和为某一值的路径

2017-04-28 22:38 405 查看
题目描述

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

解:

带记忆的DFS

/*
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>> result;
if(!root)
return result;
vector<int> trace;
find(root, expectNumber, trace, result);
return result;
}
void find(TreeNode* root, int expectNumber, vector<int>& trace, vector<vector<int>>& result){
trace.push_back(root->val);
if(!root->left && !root->right){
if(root->val == expectNumber)
result.push_back(trace);
}
if(root->left){
find(root->left, expectNumber - root->val, trace, result);
}
if(root->right){
find(root->right, expectNumber - root->val, trace, result);
}
trace.pop_back();
}
};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  面试题 二叉树 dfs