您的位置:首页 > Web前端

剑指Offer(34)二叉树中和为某一值的路径

2018-01-28 10:21 489 查看
题目描述

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

算法:DFS

数据结构:二叉树

编程语言:C++

/*
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>> ret;//返回的二维数组
vector<int> trace;//用于存储符合条件的路径
if(root)//非空判断
dfs(root,expectNumber,ret,trace);
return ret;
}
void dfs(TreeNode* root,int s,vector<vector<int>> &ret,vector<int> &trace)
{
trace.push_back(root->val);//存储各个节点的值
if(!root->left&&!root->right)//到达叶子节点
{
if(s==root->val)//如果最后一个节点的值,说明整个路径的总和与给定的值相同
ret.push_back(trace);
}
if(root->left)//向左遍历
dfs(root->left,s-root->val,ret,trace);
if(root->right)
dfs(root->right,s-root->val,ret,trace);
//如果当前的路径值不对,删除最后一个元素,向上遍历
trace.pop_back();
}
};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: