145. Binary Tree Postorder Traversal(二叉树的后序遍历,C++,递归+迭代)

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

Example:

Input: [1,null,2,3]

   1
    \
     2
    /
   3

Output: [3,2,1]
Follow up: Recursive solution is trivial, could you do it iteratively?

思路:
递归版本不说了,直接上代码非常简单。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<int> ans;
    vector<int> postorderTraversal(TreeNode* root) {
        postorder(root);
        return ans;
    }
    void postorder(TreeNode* root){
        if(root == NULL)
            return;
        postorder(root->left);
        postorder(root->right);
        ans.push_back(root->val);
    }
};

结果:
在这里插入图片描述

迭代做法: 和先序遍历和中序遍历一样,首先需要找到左下角的节点,用栈在找左下角元素的时候记录下路径上经过的节点。这段代码可以写成如下(记做代码段1):

while(p){//p初始为root节点
    stk.push(p);
    p = p->left;
}

经过代码段1后得到p==NULL,这时是访问到了下面这两种情况,最近保存的是最后入栈的图片来源
在这里插入图片描述
到达左下角后,假设此时栈顶元素记为plast左下角元素的左孩子必为NULL,但是其右孩子有两种情况,一种是其右孩子也为NULL(上图a),这样我们可以直接访问栈顶元素后让栈顶元素出栈即可。另一种情况略微复杂,其右孩子不为NULL(上图b),这时我们并不能直接访问栈顶元素,因为后序遍历的条件下,根节点应该在其右子树访问完之后访问,所以我们应该先访问栈顶元素的右子树(注意此时并没有让栈顶元素plast出栈,因为plast还没访问),假设当前栈顶元素plast的右子树访问完(栈可能经过一系列push和pop此时栈顶元素回到最初的plast节点),栈顶元素还是plast时,这时我们应该访问plast->val,并让栈顶元素出栈,而这个操作的关键条件是plast的右孩子访问完,而我们用什么来判断当栈顶元素是第一次访问,还是栈顶元素的右子树已经访问过我们可以设置一个变量lastVisited指向最后被访问的节点,当栈顶元素第一次访问时有条件:lastVisited!=plast->right而当plast的右子树访问过之后,也就是plast->right==lastVisited的时候就是我们要执行上面所说的访问plast并让其出栈的条件!

上面这些步骤完成了对一棵子树的遍历,只需要将上面的步骤循环起来就可以完成对整棵二叉树的遍历。

呼,表达能力太差了,尽力写了半天,还是写的很啰嗦。。。
talk is cheap show me the code! 如果上面说的没明白建议先看代码,之后再返回看上面这段解释,多看看应该就能理解了。

C++迭代做法代码:

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<int> ans;
    vector<int> postorderTraversal(TreeNode* root) {
        stack<TreeNode*> stk;
        TreeNode * p = root;
        TreeNode * lastVisited = NULL;//记录上一次访问过的节点
        
        if(p == NULL)
            return ans;
        
        while(!stk.empty() || p){//每次访问完都会出栈,并且设置p=NULL
            if(p){//不断往左孩子迭代,直到左孩子为空,同时路径上的节点入栈
                stk.push(p);
                p = p->left;
            }
            else{//此时p==NULL,到达了左下角节点
                p = stk.top();
                if(p->right == NULL || p->right == lastVisited){//右孩子为空,或者右孩子上次已经访问完
                    ans.push_back(p->val);//这时可以访问栈顶元素
                    lastVisited = p;//更新lastVisited
                    stk.pop();//出栈
                    p = NULL;//上面的操作表示p指向的节点已经访问完,因为是后序遍历,
                    //此时p所代表的的子树一定也访问完了,需要将指针回溯指向栈顶,访问后序元素
                    //设置p=NULL,可以让下一次p直接进入当前if外层的else程序段继续对栈顶元素进行访问
                }
                else{
                    p = p->right;//这时p是第一次被访问,不能直接访问其指向的值,应该先访问右子树
                }
            }
        }
        
        return ans;
    }
    
};

结果:
花了好大的力气搞出迭代的做法告诉我没有递归快???excuse me??? 不明觉厉,不是很懂为什么慢了。。。
在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值