题目描述
输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
解析
1.第一步,通过使用递归的深度优先遍历(DFS)将所有的根结点到叶子节点的路径放在二维数组结构中(vector<vector<int>>)。
使用了两个标记,length用来记录路径的条数。
在每次到达叶子结点的过程中,依次将结点值存入数组的一行中。
第二个标记top用来记录当前路径的深度,方便在创建分支路径时复制到达分支点前面的路径
2.找出将二维数组中路径和等于expectNumber的数组
使用冒泡排序将数组长度降序排列
void method(TreeNode* p, vector<vector<int>> &temp, int &top, int& length){
temp[length].push_back(p->val);
if(p->left){
method(p->left,temp,++top,length);
if(p->right){
vector<int> t;
cout<<top<<endl;
t.assign(temp[length].begin(), temp[length].begin()+top+1);
temp.push_back(t);
method(p->right, temp, ++top, ++length);
}
}else if(p->right)
method(p->right, temp, ++top, length);
--top;
}
int sum(vector<int> temp){
int s = 0;
for(auto a : temp){
s+=a;
}
return s;
}
vector<vector<int>> FindPath(TreeNode* root,int expectNumber) {
vector<vector<int>> temp,result;
if(!root) return result;
int top = 0;
int length = 0;
vector<int> t;
temp.push_back(t);
method(root,temp,top,length);
for(int i=0;i<temp.size();++i){
if(sum(temp[i]) == expectNumber)
result.push_back(temp[i]);
}
for(int i=0;i<result.size();i++){
for(int j=0;j<result.size()-i-1;++j){
if(result[j].size()<result[j+1].size()){
vector<int> x = result[j];
result[j] = result[j+1];
result[j+1] = x;
}
}
}
return result;
}
方法二
使用栈来完成非递归实现。具体实现比较麻烦,主要使用了下面栈的深度优先遍历的第一种途径可以找到所有根节点到叶子节点的路径,之后和方法一一样完成即可。
使用栈进行后序深度优先遍历以下有两种途径。
如下参考了https://www.cnblogs.com/SHERO-Vae/p/5800363.html后序遍历非递归实现的方法
途径一:
根节点入栈。
循环:寻找当前栈顶的最左结点,每次寻找都入栈,如果是第一次访问该最左结点,则其右结点入栈(此时,若无右结点,则当前栈中元素就是根节点到叶子节点的路径);如果是第二次访问该结点,则出栈。返回循环开始。栈为空时结束循环。
途径二:
根节点入栈。
循环:每次访问栈顶结点是否有左右结点,先右后左的方式入栈,若无左右结点或者左右结点已经被访问过,则访问当前栈顶结点,并出栈。返回循环开始。栈为空时结束循环。