LeetCode 145 二叉树后序遍历 递归非递归 通俗易懂的万能模板方法
二叉树的遍历方式作为面试热门考点,相信递归的方式大家都很容易能够写出来,但只掌握递归的方法是不够的,还需要的掌握非递归的方式,介绍一种能够解决前、中、后三种遍历的非递归形式的模板。一次性解决非递归问题!!
1.递归 二叉树后序遍历
后序遍历的顺利为 左子树,右子树,根节点。递归主要就是递归结构和递归结束条件。
递归结构:那么递归的顺序就是 先遍历左子树,遍历右子树,保存根节点。
递归结束条件:因为没有明确的结束条件,因此递归条件就是遇到NULL,返回即可。
class Solution {
public:
vector<int> preorderTraversal(TreeNode* root) {
vector<int> res;
func(root,res);
return res;
}
void func(TreeNode* root, vector<int>& res)
{
if(root==NULL) 递归结束条件
return ;
func(root->left,res); 遍历左子树
func(root->right,res); 遍历右子树
res.push_back(root->val); 保存根节点
}
};
2.非递归 二叉树后序遍历 万能模板法
对于非递归法,难度会高一些,我们将递归的方式通过使用栈的数据结构来模拟递归。
对于万能模板法,同样也是借用栈的数据结构,同时我们引入了一个类,来提供额外的信息。
我们引入了Guide类。
class Guide{
public:
int ope;
TreeNode* node;
Guide(int ope,TreeNode* node):ope(ope),node(node){}
};
我们在使用栈的过程,不存储树节点,而是存储Guide类的对象,Guide类对象包含了树节点以及额外的信息(int ope),实际上ope只有两种取值,0和1。
ope = 0代表了我们首次遍历到了某个树节点(我们首次遍历的某个节点会有三种类型的操作)
- 将当前节点的值存入最后结果中 (也就是将ope赋值为1)。
- 遍历左子树。
- 遍历右子树。
对于后序遍历是左,右,根。那也存入栈中顺序为根,右,左。对应的核心代码
st.push(Guide(1,root));
st.push(Guide(0,root->right));
st.push(Guide(0,root->left));
ope = 1代表了我们将此对象包含的树节点的值存入到最终的vector中。
那么最终的完整版代码如下。对于前序,中序非递归遍历的只需要改动上述三条语句的顺序。
class Guide{
public:
int ope;
TreeNode* node;
Guide(int ope,TreeNode* node):ope(ope),node(node){}
};
class Solution {
public:
vector<int> postorderTraversal(TreeNode *root) {
if(root == NULL)
return {};
stack<Guide> st;
st.push(Guide(0,root));
vector<int> res;
while(!st.empty())
{
Guide s = st.top();
st.pop();
if(s.node == NULL)
continue;
if(s.ope == 1)
res.push_back(s.node->val);
else{
st.push(Guide(1,root));
st.push(Guide(0,root->right));
st.push(Guide(0,root->left));
}
}
return res;
}
};