Day47 二叉树的中序遍历

给定一个二叉树的根节点 root ,返回它的 中序 遍历。

https://leetcode-cn.com/problems/binary-tree-inorder-traversal/

递归算法很简单,你可以通过迭代算法完成吗

示例1:

输入:root = [1,null,2,3]
输出:[1,3,2]

示例2:

输入:root = []
输出:[]

示例3:

输入:root = [1]
输出:[1]

示例4:

输入:root = [1]
输出:[1]

示例 5:

输入:root = [1]
输出:[1]

提示:

输入:root = [1]
输出:[1]

Java解法

思路:

  • 利用递归可以简单的完成前中后遍历

  • 还可以通过栈来处理,//中序 先加左 再自己 再右

    • 栈是先进后出结构,所以对应处理就是 先将右节点入栈

      public static List inorderTraversal2(TreeNode root) {
          List integers = new ArrayList<>();
          Stack stack = new Stack<>();
          TreeNode curr = root;
          while (curr != null || !stack.isEmpty()) {
              while (curr != null) {
                  stack.push(curr);
                  curr = curr.left;
              }
              curr = stack.pop();
              integers.add(curr.val);
              curr = curr.right;
          }
          return integers;
      }
      
      image
package sj.shimmer.algorithm.y2021;

import java.util.ArrayList;
import java.util.List;

import sj.shimmer.algorithm.TreeNode;


/**
 * Created by SJ on 2021/3/13.
 */

class D47 {
    public static void main(String[] args) {
//        TreeNode root = TreeNode.getInstance(new int[]{1, 2, 3, 4, 5});
        TreeNode root = TreeNode.getInstance(new int[]{1,2});
        System.out.println(inorderTraversal(root,0));
        System.out.println(inorderTraversal(root,1));
        System.out.println(inorderTraversal(root,2));
    }
    public static List inorderTraversal(TreeNode root, int i) {
        List integers = new ArrayList<>();
        inorderTraversal(integers,root,i);
        return integers;
    }

    public static void inorderTraversal(List integers, TreeNode node,int i){
        if (node == null) {
            return;
        }
        if (i==0) {
            //前序 先加自己 再左 再右
            integers.add(node.val);
            inorderTraversal(integers,node.left,0);
            inorderTraversal(integers,node.right,0);
        }else if (i==1){
            //中序 先加左 再自己 再右
            inorderTraversal(integers,node.left,1);
            integers.add(node.val);
            inorderTraversal(integers,node.right,1);
        }else {
            //后序 先加右 再左 再自己
            inorderTraversal(integers,node.left,2);
            inorderTraversal(integers,node.right,2);
            integers.add(node.val);
        }

    }
}
image

官方解

https://leetcode-cn.com/problems/binary-tree-inorder-traversal/solution/er-cha-shu-de-zhong-xu-bian-li-by-leetcode-solutio/

  1. 递归

    同上

    • 时间复杂度:O(n)
    • 空间复杂度:O(n)
  2. 迭代

    递归的时候隐式地维护了一个栈,而我们在迭代的时候需要显式地将这个栈模拟出来,其他都相同

    class Solution {
        public List inorderTraversal(TreeNode root) {
            List res = new ArrayList();
            Deque stk = new LinkedList();
            while (root != null || !stk.isEmpty()) {
                while (root != null) {
                    stk.push(root);
                    root = root.left;
                }
                root = stk.pop();
                res.add(root.val);
                root = root.right;
            }
            return res;
        }
    }
    

你可能感兴趣的:(Day47 二叉树的中序遍历)