leetcode题解:Binary Tree Postorder Traversal (二叉树的后序遍历)

题目：

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 };

二、非递归

根据后序遍历的顺序，先访问左子树，再访问右子树，后访问根节点，而对于每个子树来说，又按照同样的访问顺序进行遍历，后序遍历的非递归的实现相对来说要难一些，要保证根节点在左子树和右子树被访问后才能访问，思路如下：

对于任一节点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             else
33             {
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 };

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;  //若为空树，则返回空vector
16         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 };

leetcode题解:Binary Tree Postorder Traversal (二叉树的后序遍历),布布扣,bubuko.com