需要注意的点: 后序遍历与中序遍历的区别是,当访问某个节点是,可以确定其左孩子已经被访问过了,但是右孩子是否被访问过了还无法确定,因此需要使用一个prev变量,记录一下前一个被访问过的节点,如果右孩子是这个已经被访问过的节点,或者右孩子为空时,此时才能访问当前节点。(访问过该节点的含义是,将该节点放入了res数组,对于后序遍历,其必要条件是,该节点的左孩子和右孩子都被访问过了)
class Solution {
public:
vector<int> postorderTraversal(TreeNode* root) {
stack<TreeNode*> st;
vector<int> res;
TreeNode* prev = nullptr;
while (root || !st.empty()) {
while (root) {
st.push(root);
root = root->left;
}
TreeNode* tmp = st.top();
st.pop();
if (!tmp->right || tmp->right == prev) {
res.emplace_back(tmp->val);
prev = tmp;
root = nullptr;
} else {
st.push(tmp);
root = tmp->right;
}
}
return res;
}
};
|