Leetcode 94. Binary Tree Inorder Traversal

来源:互联网 发布:测量图片尺寸软件 编辑:程序博客网 时间:2024/04/30 18:52


递归的方法:

public class Solution {    public ArrayList<Integer> inorderTraversal(TreeNode root) {        ArrayList<Integer> list = new ArrayList<Integer>();        if(root != null){            list.addAll(inorderTraversal(root.left));            list.add(root.val);            list.addAll(inorderTraversal(root.right));        }        return list;    }}



非递归方法:

public class Solution {    public ArrayList<Integer> inorderTraversal(TreeNode root) {        ArrayList<Integer> list = new ArrayList<Integer>();        Stack<TreeNode> s = new Stack<TreeNode>();        while(!s.isEmpty() || root!=null){            if(root!=null){                s.push(root);                root = root.left;            }            else{                root = s.pop();                list.add(root.val);                root = root.right;            }        }        return list;    }}

第三种方法Morris Traversal,空间复杂度为O(1):


步骤:

1. 如果当前节点的左孩子为空,则输出当前节点并将其右孩子作为当前节点。

2. 如果当前节点的左孩子不为空,在当前节点的左子树中找到当前节点在中序遍历下的前驱节点。

   a) 如果前驱节点的右孩子为空,将它的右孩子设置为当前节点。当前节点更新为当前节点的左孩子。

   b) 如果前驱节点的右孩子为当前节点,将它的右孩子重新设为空(恢复树的形状)。输出当前节点。当前节点更新为当前节点的右孩子。

3. 重复以上1、2直到当前节点为空。


public ArrayList<Integer> inorderTraversal(TreeNode root) {      ArrayList<Integer> res = new ArrayList<Integer>();      TreeNode cur = root;      TreeNode pre = null;      while(cur != null)      {          if(cur.left == null)          {              res.add(cur.val);              cur = cur.right;          }          else          {              pre = cur.left;              while(pre.right!=null && pre.right!=cur)                  pre = pre.right;              if(pre.right==null)              {                  pre.right = cur;                  cur = cur.left;              }              else              {                  pre.right = null;                  res.add(cur.val);                  cur = cur.right;              }          }      }      return res;  }  



复杂度分析:

空间复杂度:O(1),因为只用了两个辅助指针。

时间复杂度:O(n)。证明时间复杂度为O(n),最大的疑惑在于寻找中序遍历下二叉树中所有节点的前驱节点的时间复杂度是多少,即以下两行代码:

1 while (prev->right != NULL && prev->right != cur)2     prev = prev->right;

直觉上,认为它的复杂度是O(nlgn),因为找单个节点的前驱节点与树的高度有关。但事实上,寻找所有节点的前驱节点只需要O(n)时间。n个节点的二叉树中一共有n-1条边,整个过程中每条边最多只走2次,一次是为了定位到某个节点,另一次是为了寻找上面某个节点的前驱节点,如下图所示,其中红色是为了定位到某个节点,黑色线是为了找到前驱节点。所以复杂度为O(n)。




Morris Traversal 详见网上的一篇博客 http://www.cnblogs.com/AnnieKim/archive/2013/06/15/MorrisTraversal.html









0 0
原创粉丝点击