title: 2019-8-21 二叉树中和为某一值的路径
tags: 算法,每日一题,二叉树
二叉树中和为某一值的路径
1. 题目描述
输入一颗二叉树的根节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意:在返回值的list中,数组长度大的数组靠前)
2.题目解析
2.1 思路解析
这题要注意两点:1)寻找的路径是从根节点到叶子节点的路径,不用考虑中间路径的情况;2)输出结果最长的路径在最前面。
方法一:使用深度优先的方式(对于二叉树可以使用先序遍历)遍历二叉树,得到每一条从根节点到叶子节点的路径,找出符合条件的路径。如何使得结果满足第二个条件呢?找完之后再重新排序,可能有其它更好的方法,大家有想到的麻烦留言告知。
方法二:还是使用先序遍历的方式,当到达某一深度之后走过的节点的累加和大于目标值了,那么之后的节点就不用再访问了,直接退出。否则继续到叶子节点看路径是否符合。(这种方式不可以,因为节点的值可以是负数)
/*
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<int> trace;
vector<vector<int> > res;
if (root) dfsFindPath(root, expectNumber, trace, res);//要先判断root不为空
return res;
}
void dfsFindPath(TreeNode* root, int expectNumber, vector<int> &trace, vector<vector<int> >& res){
trace.push_back(root->val);
if(root->left == NULL && root->right == NULL){//叶子节点
if(expectNumber == root->val){
res.push_back(trace);
}
}else{//当前节点为非叶子节点
if (root->left) dfsFindPath(root->left, expectNumber - root->val, trace, res);
if (root->right) dfsFindPath(root->right, expectNumber - root->val, trace, res);
}
if (!trace.empty()) trace.pop_back();//回退
}
};
更多关于编程和机器学习资料请关注FlyAI公众号。