给定一个二叉树的根节点 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<Integer> inorderTraversal2(TreeNode root) { List<Integer> integers = new ArrayList<>(); Stack<TreeNode> 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; }
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<Integer> inorderTraversal(TreeNode root, int i) {
List<Integer> integers = new ArrayList<>();
inorderTraversal(integers,root,i);
return integers;
}
public static void inorderTraversal(List<Integer> 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);
}
}
}
官方解
-
递归
同上
- 时间复杂度:O(n)
- 空间复杂度:O(n)
-
迭代
递归的时候隐式地维护了一个栈,而我们在迭代的时候需要显式地将这个栈模拟出来,其他都相同
class Solution { public List<Integer> inorderTraversal(TreeNode root) { List<Integer> res = new ArrayList<Integer>(); Deque<TreeNode> stk = new LinkedList<TreeNode>(); 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; } }