题目描述
输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
思路:
根据题目知,每一条路径都是从根节点一直到叶子节点,我们可以想到深度优先遍历,根节点肯定是在路径最开始的,所以我们可以采用先序的深度优先遍历。最先访问根节点,下来是左子树,接着才是左子树,根节点>左子树>右子树。当一次遍历完成后,如果经过的路径的和刚好等于输入的数值,则将路径输出并回退一个节点(回到父节点);若经过的路径的和不等于输入的值,则直接回退一个节点。当当前的父节点还有右子树时,继续遍历,否则,继续回退。一直到根节点的左右子树都遍历完成后,此时结束。
以上述的树为例:假设输入的数为三
第一次:d b a ,1+1+1=3,输出路径,回退到b节点上。
第二次:d b c,1+1+1 =3,输出路径,回退到b节点上,但b节点的左右子树都已经访问过,因此回退到d节点上。
第三次:d 2,1+2=3,输出路径,回退到d节点上,d节点左右子树都已访问,则遍历结束。
实现:C++
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
vector<vector<int>>pathlist;
vector<int>path;
public:
vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
if(root==NULL)
{
return pathlist;
}
path.push_back(root->val);
if(root->left==NULL&&root->right==NULL&&root->val==expectNumber)
{
pathlist.push_back(path);
}
if(root->val<=expectNumber&&root->left!=NULL)//当前根节点有左子树
{
FindPath(root->left,expectNumber-root->val);
}
if(root->val<=expectNumber&&root->right!=NULL)//访问右子树
{
FindPath(root->right,expectNumber-root->val);
}
//访问都完成后,回退到父节点
path.pop_back();
return pathlist;
}
};
主要也是用到了递归的思想.