leetcode題解:Binary Tree Postorder Traversal (二叉樹的後序遍曆)

標籤：des   style   class   blog   code   http   

題目：

Given a binary tree, return the postorder traversal of its nodes‘ values.

For example:
Given binary tree {1,#,2,3},

   1         2    /   3

 

return [3,2,1].

Note: Recursive solution is trivial, could you do it iteratively?

說明:

      1) 兩種實現，遞迴與非遞迴 , 其中非遞迴有兩種方法

      2）複雜度分析：時間O(n)、空間O(n)

 

實現：

一、遞迴

 1 /** 2  * Definition for binary tree 3  * struct TreeNode { 4  *     int val; 5  *     TreeNode *left; 6  *     TreeNode *right; 7  *     TreeNode(int x) : val(x), left(NULL), right(NULL) {} 8  * }; 9  */10 class Solution {11 public:12     vector<int> postorderTraversal(TreeNode *root) {13         vector<int> root_vec;14         vector<int> left_vec;15         vector<int> right_vec;16         if(root==NULL) return root_vec;17         if(root->left) left_vec=postorderTraversal(root->left);18         if(root->right) right_vec=postorderTraversal(root->right);19         root_vec.push_back(root->val);20         left_vec.insert(left_vec.end(),right_vec.begin(),right_vec.end());21         left_vec.insert(left_vec.end(),root_vec.begin(),root_vec.end());22         return left_vec;23     }24 };

View Code

二、非遞迴

根據後序遍曆的順序，先訪問左子樹，再訪問右子樹，後訪問根節點，而對於每個子樹來說，又按照同樣的訪問順序進行遍曆，後序遍曆的非遞迴的實現相對來說要難一些，要保證根節點在左子樹和右子樹被訪問後才能訪問，思路如下：

 

對於任一節點P，

1）先將節點P入棧；

2）若P不存在左孩子和右孩子，或者P存在左孩子或右孩子，但左右孩子已經被輸出，則可以直接輸出節點P，並將其出棧，將出棧節點P標記為上一個輸出的節點，再將此時的棧頂結點設為當前節點；

3）若不滿足2）中的條件，則將P的右孩子和左孩子依次入棧，當前節點重新置為棧頂結點，之後重複操作2）；

4）直到棧空，遍曆結束。

a、下面代碼比較常規，與上面分析思路一致

 1 /** 2  * Definition for binary tree 3  * struct TreeNode { 4  *     int val; 5  *     TreeNode *left; 6  *     TreeNode *right; 7  *     TreeNode(int x) : val(x), left(NULL), right(NULL) {} 8  * }; 9  */10 class Solution {11 public:12     vector<int> postorderTraversal(TreeNode *root) {13         vector<int> postorder_vec;14         TreeNode *cur=root; //定義指標，指向當前節點  15         TreeNode *pre=NULL;//定義指標，指向上一各訪問的節點   16         if(cur==NULL) return postorder_vec;17         stack<TreeNode *> postorder_stack;//建立一個空棧  18         postorder_stack.push(cur);//先將樹的根節點入棧19         //直到棧空時，結束迴圈20         while(!postorder_stack.empty())21         {22             cur=postorder_stack.top();//當前節點置為棧頂節點  23             if((cur->left==NULL&&cur->right==NULL)||24             ((pre!=NULL)&&(cur->left==pre||cur->right==pre)))25             {26             //如果當前節點沒有左右孩子，或者有左孩子或有孩子，但已經被27             //訪問輸出,則直接輸出該節點，將其出棧，將其設為上一個訪問的節點  28                 postorder_stack.pop();29                 postorder_vec.push_back(cur->val);30                 pre=cur;31             }32             else33             {34                  //如果不滿足上面兩種情況,則將其右孩子左孩子依次入棧  35                 if(cur->right!=NULL) postorder_stack.push(cur->right);36                 if(cur->left!=NULL) postorder_stack.push(cur->left);37             }38         }39     }40 };

View Code

b、下面代碼簡潔（個人感覺，不喜勿噴）

 1 /** 2  * Definition for binary tree 3  * struct TreeNode { 4  *     int val; 5  *     TreeNode *left; 6  *     TreeNode *right; 7  *     TreeNode(int x) : val(x), left(NULL), right(NULL) {} 8  * }; 9  */10 class Solution {11 public:12    vector<int> postorderTraversal(TreeNode *root)   13     {  14         vector<int> rs;  15         if (!root) return rs;  //若為空白樹，則返回空vector16         stack<TreeNode *> stk;  17         stk.push(root);  //當前節點入棧18         while (!stk.empty())  19         {  20             TreeNode *t = stk.top();  //棧頂節點出棧、輸出21             stk.pop();  22             rs.push_back(t->val);  23             //注意，下面入棧順序不能錯 ，因為先左後右，24             //這樣出棧時先遍曆才是右（中->右->左）25             if (t->left) stk.push(t->left);  26             if (t->right) stk.push(t->right);  27         }  28         reverse(rs.begin(), rs.end());  //逆序，就成了後序遍曆了29         return rs;  30     }  31 };

View Code