给一个二叉树,用栈模拟递归,返回二叉树的后续遍历(postorder)。
/**
* 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> answer;
stack<TreeNode*> st;
vector<int> postorderTraversal(TreeNode* root) {
TreeNode* prve = nullptr;
while(!st.empty()||root!=nullptr){
while(root!=nullptr){
st.push(root);
root = root->left;
}
root = st.top();
st.pop();
if(root->right==nullptr||root->right==prve)
{
answer.push_back(root->val);
prve = root;
root = nullptr;
}
else {
st.push(root);
root = root->right;
}
}
return answer;
}
};
思路
大while里面套着一个小while。
小while作用是向左遍历,将路过元素压入栈,当遇到null后,将栈顶提取,观察它的右树若满足,右树为空或之前已经被push进答案里了,说明此root满足被push到答案的条件,将之前prve更新为刚被push的root,将root替换为空,这样就不用经历小while,直接继续对栈顶进行观察,若发现它既不满足右树为空,也不满足右树为prve,说明现在还不能push它到答案,因为它还有右子树,继续把root恢复到栈,再root = root ->right,继续向右方向push到栈。
最后大while终止的条件需要满足栈空,且root空。