二叉树的后序遍历
给定一个二叉树,返回它的 后序 遍历。
示例:
输入: [1,null,2,3]
1
\
2
/
3
输出: [3,2,1]
进阶: 递归算法很简单,你可以通过迭代算法完成吗?
/** * 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> postorderTraversal(TreeNode* root) { vector<int> res; if(root == nullptr){ return res; } stack<TreeNode *> stk; TreeNode *pre = nullptr; while(root != nullptr || !stk.empty()){ while(root != nullptr){ stk.emplace(root); root = root->left; } root = stk.top(); stk.pop(); if(root->right == nullptr || root->right == pre){//右子树已经被遍历完或者右子树为空 res.emplace_back(root->val); pre = root; root = nullptr;//为了避开上面的循环,回溯完成右子树 } else{ stk.emplace(root);//右子树不空的话,该节点再次入栈,转而遍历右子树 root = root->right; } } return res; } };
1.主要练习迭代算法
2.借助一个栈来保留之前的节点,一开始直接访问到左子树的末端,然后往回回溯
3.如果右子树为空或者说右子树已经被访问结束,则该节点已经使用完毕开始弹出往外回溯
4.注意弹出后root赋值为nullptr来保证程序的正常运行