【LeetCode】LeetCode刷题(19)【简单】二叉树的前&&中&&后遍历(非递归)(C++)
@TOC
精华在于进栈和出栈的时机
94.二叉树的中序遍历
思路: 中序遍历的顺序是,左 - 根 - 右 创建一个栈来存储结点,创建一个vector来存储中序遍历的值 从根结点开始,只要该结点有左子树,就将该结点压进栈中。 直到root为空。 取出栈顶元素,栈顶元素出栈,将该结点值存进recv。 ... 剩下的只可意会不可言传了,
感谢这位老哥分享——链接
class Solution {
public:
//中序遍历顺序-左-中-右
vector<int> inorderTraversal(TreeNode* root) {
vector<int>recv;
stack<TreeNode*> Tstack;
//当前结点不为空或当前栈不为空
while(root || !Tstack.empty())
{
while(root)
{
//只要当前结点不为空就往栈里面压
Tstack.push(root);
root = root->left;
}
//此时栈顶元素为根节点左侧树最左的左子树
//取到该结点
root = Tstack.top();
Tstack.pop();
//pop出栈,存进recv中
recv.push_back(root->val);
root = root->right;
}
return recv;
}
};
递归方法
144.二叉树的前序遍历
class Solution {
public:
vector<int> preorderTraversal(TreeNode* root) {
vector<int>recv;
stack<TreeNode*>Tstack;
while(root || !Tstack.empty())
{
while(root)
{
recv.push_back(root->val);
Tstack.push(root);
root = root->left;
}
root = Tstack.top();
Tstack.pop();
root = root->right;
}
return recv;
}
};
145.二叉树的后序遍历
一直往栈里面往左节点,压到左边最后一个做结点,往回pop,判断当前这个结点是否右结点,有右结点就输出,最后判断自己。
感谢这位老哥分享思路—链接
class Solution { public: vector<int> postorderTraversal(TreeNode* root) { vector<int>result; stack<TreeNode*>Tstack; TreeNode* cur = root; TreeNode* prev = nullptr;//记录cur上一个指向的结点,比cur走慢一步
while(!Tstack.empty() || cur)
{
//只有cur不为空,就一直往里面压左节点
while(cur)
{
Tstack.push(cur);
cur =cur->left;
}
cur = Tstack.top();
//如果当前结点没有右结点 || 右结点已经访问过了
if(!cur->right || prev == cur->right)
{
Tstack.pop();
result.push_back(cur->val);
prev = cur;
//要从栈里面往外面吐结点,所以要将cur置为null
cur = nullptr;
}
else
{
cur = cur->right;
}
}
return result;
}
};
大致流程感觉
![在这里插入图片描述](https://img-blog.csdnimg.cn/88d1252a9a964302bf56a25c8757cb2c.jpg?x-oss-process=image/watermark,type_ZHJvaWRzYW5zZmFsbGJhY2s,shadow_50,text_Q1NETiBARG8yZU0wTg==,size_20,color_FFFFFF,t_70,g_se,x_16#pic_center)![在这里插入图片描述](https://img-blog.csdnimg.cn/be98cf30ed3240ceabe2540a448efaff.gif#pic_center#pic_center)