1- Description
输入一颗二叉树的根节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
struct TreeNode { int val; struct TreeNode *left; struct TreeNode *right; TreeNode(int x) : val(x), left(NULL), right(NULL) { } };
2-Solution
首先要明确的是路径具有以树的根结点开始到叶结点结束的特点,而前序遍历正是从根结点开始的遍历。我么可以以下图的二叉树为例,为找到和为22的路径进行过程分析。
很显然的可以看出这里有两条符合要求的路径(10 - 5 - 7和10 - 12),而具体的寻找过程可以简单用下边表示:
步骤 | 操作 | 是否叶结点 | 路径 | 路径结点和 |
---|---|---|---|---|
1 | 访问结点10 | 否 | 10 | 10 |
2 | 访问结点5 | 否 | 10 - 5 | 15 |
3 | 访问结点4 | 是 | 10 - 5 - 4 | 19 |
4 | 回到结点5 | 否 | 10 - 5 | 15 |
5 | 访问结点7 | 是 | 10 - 5 - 7 | 22 |
6 | 回到结点5 | 否 | 10 - 5 | 15 |
7 | 回到结点10 | 否 | 10 | 10 |
8 | 访问结点12 | 是 | 10 - 12 | 22 |
从上述过程可以分析得到如下算法步骤:
1-利用前序遍历访问二叉树的结点,并将结点添加到路径中,同时累加结点值
2-如果该结点为叶子结点,且结点值和期望值相等,则该路径符合要求;如果该结点非叶子结点,则继续访问其子节点
3-当结点访问结束后,递归函数将自动回到其父节点,注意在回退过程中删除当前结点,以保证返回父节点是路径刚好是从根结点到父节点。
根据以上算法步骤可以写成如下代码:
class Solution {
public:
vector<vector<int> > respath;
vector<int > path;
void FindSubPath(TreeNode* root, int expectNumber){
if(root == NULL) return;//如果树为空直接返回空
path.push_back(root->val);//存入路径结点值
expectNumber -= root->val;//相应减去结点值
bool bisleaf = !root->left && !root->right;//确认是叶子结点
if(bisleaf && expectNumber == 0){//如果遍历完到叶结点,且和为期望值
respath.push_back(path);//存路径
}
FindSubPath(root->left, expectNumber);//遍历左子树
FindSubPath(root->right, expectNumber);//遍历右子树
path.pop_back();//递归完成一次结点回退一次,即回到父结点
}
vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
FindSubPath(root,expectNumber);
return respath;
}
};