Leetcode145.二叉数的后序遍历

Posted 2021-02-21 cat-fish-

145.二叉数的后序遍历

给定一个二叉树，返回它的 后序 遍历。

示例:

输入: [1,null,2,3]
1
??
???2
?? /
3

输出: [3,2,1]

进阶: 递归算法很简单，你可以通过迭代算法完成吗？

后序遍历的概念

后序遍历就是先遍历左子树，然后遍历右子树，然后访问根节点。

递归实现

先左子树,然后右子树，最后根

vector<int> v;
    vector<int> inorderTraversal(TreeNode* root) {
        if(root != NULL) {
        inorderTraversal(root->left);
        inorderTraversal(root->right);
        v.push_back(root->val);
        }
        return v;
    }

迭代实现

方法一

后序遍历相对前序，中序要麻烦不少

其中一种方法是我们通过观察发现，后序遍历与前序遍历相对称。

思路： 每到一个节点 A，就应该立即访问它。 然后将左子树压入栈，再次遍历右子树。

遍历完整棵树后，结果序列逆序即可。

vector<int> postorderTraversal(TreeNode* root) {
        stack<TreeNode*> s;
        vector<int> v;
        TreeNode* rt = root;
        while(rt || s.size()){
            while(rt){
                s.push(rt->left);
                v.push_back(rt->val);
                rt=rt->right;
            }
            rt=s.top();s.pop();
        }
        reverse(v.begin(),v.end());
        return v;
    }

方法二

在看了力扣上jason-2的题解后不得不说通过栈的方式确实很巧妙

思路：每到一个节点 A，因为根要最后访问，将其入栈。然后遍历左子树，遍历右子树，最后返回到 A。

但是出现一个问题，无法区分是从左子树返回，还是从右子树返回。

因此，给 A 节点附加一个标记T。在访问其右子树前，T 置为 True。之后子树返回时，当 T 为 True表示从右子树返回，否则从左子树返回。

当 T 为 false 时，表示 A 的左子树遍历完，还要访问右子树。

同时，当 T 为 True 时，表示 A 的两棵子树都遍历过了，要访问 A 了。并且在 A 访问完后，A 这棵子树都访问完成了。

vector<int> postorderTraversal(TreeNode* root) {
        stack<TreeNode*> S;
        unordered_map<TreeNode*,int> done;
        vector<int> v;
        TreeNode* rt = root;
        while(rt || S.size()){
            while(rt){
                S.push(rt);
                rt=rt->left;
            }
            while(S.size() && done[S.top()]){
                v.push_back(S.top()->val);
                S.pop();
            }
            if(S.size()){
                rt=S.top()->right;
                done[S.top()]=1;    
            }
        }
        return v;
    }