JZ24_二叉树中和为某一值的路径
知识点:二叉树
题目链接
题目描述
输入一颗二叉树的根节点和一个整数,按字典序打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
示例1
输入: {10,5,12,4,7},22
返回值: [[10,5,7],[10,12]]
示例2
输入: {10,5,12,4,7},15
返回值: [ ]
解题思路
- 解决这道题 我们要知道如何遍历所有从根到叶子结点的路径
- 第一 传入的参数 和 返回值 传入最后结果数组,临时数组,结点和值
- 第二 如何返回 也就是找到叶子结点 左边右边子树都为空时返回 返回前遍历临时数组中的和 是否和值相同
- 第三 如何递归 这里采用先序遍历 先把根结点放入临时数组 向左子树找 再向右子树 注意回溯
代码
#include "cheader.h"
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>> ans;
vector<int> tmp;
if(root == nullptr)
return ans;
find(ans,tmp,root,expectNumber);
return ans;
}
void find(vector<vector<int>>&ans,vector<int>&tmp,TreeNode* root,int num){
tmp.push_back(root->val);
if(root->left == nullptr && root->right == nullptr){
int sum = 0;
for(int x:tmp)
sum+=x;
if(sum == num)
ans.push_back(tmp);
return ;
}
if(root->left){
find(ans,tmp,root->left,num);
tmp.pop_back();
}
if(root->right){
find(ans,tmp,root->right,num);
tmp.pop_back();
}
}
};
今天也是爱zz的一天!