思路
- 创建一个 List 类型的 output 变量作为输出。
- 如果传入参数 root 为 null,则直接返回 output。
- 创建一个 Stack 类型的 stack 变量作为栈,并将 root 赋值给 curr,curr 是当前节点。
- 在 while 循环中,如果 curr 不为 null,则将 curr 和它的左子树节点全部压入 stack 中,并将 curr 的值赋值为 curr.left,这步操作可以让 curr 沿着左子树一直往下遍历,直到遇到没有左子树为空的节点为止。
- 当前一轮 while 循环结束时,curr 已经遍历到了最左侧的叶子节点,因此需要弹出 stack 顶部元素(即最左侧的叶子节点);同时将该节点的 val 值存储在 output 数组中,该节点已经遍历完成,接下来需要遍历它的右子树节点,在 while 循环内部,将 curr 节点指向 curr.right 即可开始遍历右子树。
- 当所有节点都被遍历完后,while 循环退出。返回 output 数组作为中序遍历结果。
import java.util.ArrayList;
import java.util.List;
import java.util.Stack;
class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) { val = x; }
}
public class MidTree {
public static List<Integer> inorderTraversal(TreeNode root) {
// 声明返回值
List<Integer> output = new ArrayList<>();
// 边界值判断
if(root == null){
return output;
}
// 借助stack 存储左子树的值
Stack<TreeNode> stack = new Stack<>();
// 遍历root
TreeNode curr = root;
while(curr != null || !stack.isEmpty()){
while(curr != null){
// 将当前节点加入到stack中
stack.push(curr);
// 继续寻找最左节点
curr = curr.left;
}
// 节点没有左子树,所以cur是null, 此时需要将stack中的最后添加的节点找出来
curr = stack.pop();
// 将节点的值放进res中
output.add(curr.val);
// 此时应该寻找cur的右子树
curr = curr.right;
}
return output;
}
public static void main(String[] args) {
TreeNode root = new TreeNode(1);
root.left = new TreeNode(5);
root.right = new TreeNode(2);
root.right.left = new TreeNode(3);
List<Integer> result = inorderTraversal(root);
System.out.println(result); // Expected Output: [5, 1, 3, 2]
}
}
Java中Deque和Stack的区别
Java中Deque和Stack都是用于实现栈(stack)数据结构的类,但它们有一些区别。
- 接口不同
- Stack继承自Vector类,因此Stack本身是线程安全的。
- Deque是Java集合框架中定义的接口,它有多种实现,包括ArrayDeque和LinkedList等。与Stack不同,Deque实际上是一个双端队列(double-ended queue),可以在队列头部和尾部添加或删除元素。
- 实现方式不同
- Stack使用单向链表实现,在栈顶插入和删除元素时时间复杂度为O(1),但在底层数组进行扩容时时间复杂度为O(n)。
- Deque的两个实现ArrayDeque和LinkedList的实现方式不同,ArrayDeque使用循环数组实现,在队列头尾插入和删除元素时时间复杂度为O(1),但当数组需要扩容时,时间复杂度为O(n);LinkedList使用双向链表实现,在任意位置插入和删除元素时时间复杂度为O(1)。
- 用途不同
- Stack通常用于实现后进先出(LIFO)结构,例如程序中的函数调用栈、撤销操作、编辑器中的历史记录等。
- Deque既可以用于LIFO结构也可以用于FIFO(先进先出)结构,还可以用于比较复杂的数据结构,例如队列、双端队列、栈等。
综上,Stack和Deque虽然都是使用Java实现栈的类,但它们具有不同的接口、实现方式和用途,需要根据具体的情况选择合适的类。