剑指offer-24:二叉树中和为某一值的路径
目录
1题目描述
输入一颗二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
2题目答案
DFS深度搜索
BFS广度搜索
/*
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 sum,vector<vector<int>> &ret,vector<int> &trace) {
trace.push_back(root->val); //将跟节点的值放入trace数组中
if(!root->left&&!root->right) //如果左右子树均是空,且sum等于跟的值,在ret中加入一条路径
if(sum == root->val)
ret.push_back(trace);
if(root->left) //如果左子树不为空,寻找是否有符合的路径
DFS(root->left,sum-root->val,ret,trace);
if(root->right)
DFS(root->right,sum-root->val,ret,trace);
trace.pop_back(); //移除最后一个元素,深度遍历完一条路径后要回退,不然的话vector里面的元素只会增加不会减少。
}
};