要保證根結點在左孩子和右孩子訪問之後才能訪問,因此對於任一結點P,先將其入棧。如果P不存在左孩子和右孩子,則可以直接訪問它;或者P存在左孩子或者右孩子,但是其左孩子和右孩子都已被訪問過了,則同樣可以直接訪問該結點。若非上述兩種情況,則將P的右孩子和左孩子依次入棧,這樣就保證了每次取棧頂元素的時候,左孩子在右孩子前面被訪問,左孩子和右孩子都在根結點前面被訪問。
/**
* Definition for binary tree
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
vector<int> postorderTraversal(TreeNode *root) {
vector<int> result;
stack<TreeNode *> tree;
TreeNode *p=root,*pre=NULL;
if(p==NULL) return result;
tree.push(p);
while(!tree.empty()){
p=tree.top();
if((p->left==NULL && p->right==NULL)||(pre!=NULL && (pre==p->left||pre==p->right)))
{
result.push_back(p->val);
pre=p;
tree.pop();
}
else{ //注意左右孩子的入棧順序,保證左孩子在右孩子之前被訪問
if(p->right!=NULL) tree.push(p->right);
if(p->left!=NULL) tree.push(p->left);
}
}
return result;
}
};