二叉树—二叉树的后序遍历(leetcode 145)

题目描述

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

示例:

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

输出: [3,2,1]

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

算法分析

Morris 遍历

思路与算法

有一种巧妙的方法可以在线性时间内，只占用常数空间来实现后序遍历。这种方法由 J. H. Morris 在 1979 年的论文「Traversing Binary Trees Simply and Cheaply」中首次提出，因此被称为 Morris 遍历。

Morris 遍历的核心思想是利用树的大量空闲指针，实现空间开销的极限缩减。其后序遍历规则总结如下：

    新建临时节点，令该节点为 root；

    如果当前节点的左子节点为空，则遍历当前节点的右子节点；

    如果当前节点的左子节点不为空，在当前节点的左子树中找到当前节点在中序遍历下的前驱节点；

        如果前驱节点的右子节点为空，将前驱节点的右子节点设置为当前节点，当前节点更新为当前节点的左子节点。

        如果前驱节点的右子节点为当前节点，将它的右子节点重新设为空。倒序输出从当前节点的左子节点到该前驱节点这条路径上的所有节点。当前节点更新为当前节点的右子节点。

    重复步骤 2 和步骤 3，直到遍历结束。

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    void postMorris(TreeNode* root, vector<int>& ans) {
        TreeNode* cur = reverseList(root);
        while(cur) {
            ans.emplace_back(cur->val);
            cur = cur->right;
        }
    }
    TreeNode* reverseList(TreeNode* root) {
        TreeNode* cur = root;
        TreeNode* pre = nullptr;
        while(cur) {
            TreeNode* tmp = cur->right;
            cur->right = pre;
            pre = cur;
            cur = tmp;
        }
        return pre;
    }
    vector<int> postorderTraversal(TreeNode* root) {
        TreeNode* p1 = root;
        TreeNode* p2 = nullptr;
        vector<int> ans;
        while(p1) {
            if(p1->left) {
                p2 = p1->left;
                while(p2->right && p2->right != p1) {
                    p2 = p2->right;
                }
                if(p2->right == nullptr) {
                    p2->right = p1;
                    p1 = p1->left;
                    continue;
                } else {
                    p2->right = nullptr;
                    postMorris(p1->left, ans);
                }
            }
            p1 = p1->right;
        }
        postMorris(root, ans);
        return ans;
    }
};

时间复杂度分析

时间复杂度：O(n)，其中 n 是二叉树的节点数。没有左子树的节点只被访问一次，有左子树的节点被访问两次。

空间复杂度：O(1)。只操作已经存在的指针（树的空闲指针），因此只需要常数的额外空间。