LeetCode笔记整理3 二叉树中序遍历 Morris中序遍历

二叉树的中序遍历

使用栈模拟递归

var inorderTraversal = function(root) {
    var stack = []
    var res = []

    while(root || stack.length){
        while(root){
            stack.push(root)
            root = root.left
        }
        root = stack.pop()
        res.push(root.val)
        root = root.right
    }
    return res
};

Morris中序遍历 LeetCode官方

思路与算法

Morris遍历算法是另一种遍历二叉树的方法, 它能将非递归的中序遍历空间复杂度将为O(1).

Morris遍历算法整体步骤如下(假设当前遍历到的节点为X)

  1. 如果X无左孩子, 先将X的值加入答案数组, 再访问X的右孩子, 即X=X.right。
  2. 如果X有左孩子, 则找到X左子树上最右的节点(即左子树中序遍历的最后一个节点, X在中序遍历中的前驱节点), 我们记为predecessor(我称之为R)。 根据r的右孩子是否为空, 进行如下操作。
    • 如果r的右孩子为空, 则将其右孩子指向X, 然后访问X的左孩子, 即X=X.left
    • 如果r的右孩子不为空, 则此时其右孩子指向X, 说明我们已经遍历完X的左子树, 我们将R的右孩子置空, 将X的值加入答案数组, 然后访问X的右孩子, 即X=X.right
  3. 重复上述操作, 直至访问完整棵树
var inorderTraversal = function(root) {
    var res = []
    while(root){
        if(root.left){
            var r = root.left;
            while(r.right && root != r.right){
                r = r.right
            }
            if(!r.right){
                r.right = root
                root = root.left
            } 
            else {
                res.push(root.val)
                root = r.right
                root = root.right
            }
        } 
        else {
            res.push(root.val)
            root = root.right
        }
    }
    return res;
};

时间复杂度:)O(n),其中 n 为二叉搜索树的节点个数。Morris 遍历中每个节点会被访问两次,因此总时间复杂度为 O(2n)=O(n)
空间复杂度:O(1)O(1)O(1)

作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/binary-tree-inorder-traversal/solution/er-cha-shu-de-zhong-xu-bian-li-by-leetcode-solutio/
来源:力扣(LeetCode)

posted @ 2020-09-14 22:31  lemon-Xu  阅读(448)  评论(0编辑  收藏  举报