给定一个二叉树,返回它的 后序 遍历。
思路:本题考查的是后序遍历迭代版,观察后序遍历的过程可以发现,我们首先要找到最左侧的节点(可能是右孩子也可能是左孩子),并从此处开始迭代。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
vector<int> res;
vector<int> postorderTraversal(TreeNode* root) {
if(!root)
return res;
postorder(root);
return res;
}
void postorder(TreeNode* root)
{
TreeNode* p = root;
TreeNode* r = nullptr;
stack<TreeNode*> s;
while(p!=nullptr||!s.empty())
{
while(p!=nullptr)
{
s.push(p);//走到最左边(有可能是左孩子 也有可能是右孩子)
p = p->left;
}
if(!s.empty())
{
p = s.top();
if(p->right&&p->right!=r) //右子树存在 并且没有被访问
p = p->right;
else
{
s.pop();
res.push_back(p->val);
r = p;//记录最近访问过的节点
p = nullptr;//节点访问完后 重置p指针
}
}
}
}
};