leetcode:Binary Tree Postorder Traversal

       要保證根結點在左孩子和右孩子訪問之後才能訪問,因此對於任一結點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;
    }
};


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章