剑指offer:二叉树中和为某一值的路径 c/c++
题目描述
输入一颗二叉树的根节点和一个整数,按字典序打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
一些思路
根据题目要求 求路径的和 可知应该进行树的深度优先搜索遍历。
- 深度优先搜索遍历 ,深度优先搜索遍历使用的是递归的方法;
// 关于DFS遍历代码:
void DFS(TreeNode* root){
if(root==NULL) return;
cout<<root->val;
if(root->left) DFS(root->left);
if(root->right) DFS(root->right);
}
2.判断和为何值 , 每遍历一个结点,使用所要求的整数减去该结点的值,若到叶结点时,刚好该值为0,则该路径为所求路径,将此路径加入最后要返回的二维列表中。
3.,存储结果 ,首先要设置两个全局变量,一个是一维数组,用来存储每次遍历得到的路径;还有一个二维数组,用来存储符合条件的路径。
4. 对于不符合条件的路径 ,每次遍历完成一条路径后,都会将最后的节点退出数组,直到进入下一条路径;(我自己想的是判断符合条件以后,就将一维数组的内容清空,再重新加进去,最后递归结束后也清空,但是这样忽略了一个问题,下一条路径中前面的节点也是前一条路径的一些结点,因此不能完全清空,而是需要一个一个退出数组)。
完整代码
class Solution {
public:
vector<vector<int> > vec;
vector<int> path;
vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
DFS(root,expectNumber);
return vec;
}
void DFS(TreeNode* root,int expectNumber){
if(root==NULL) return;
expectNumber-=root->val;
path.push_back(root->val);
if(!expectNumber&&!root->left&&!root->right){
vec.push_back(path);
}
DFS(root->left,expectNumber);
DFS(root->right,expectNumber);
path.pop_back();
}
};