阅读量:0
题目
思路
使用一个栈来模拟递归的过程,以非递归的方式完成中序遍历(使用栈可以避免递归调用的空间消耗)。
遍历顺序步骤:
- 遍历左子树
- 访问根节点
- 遍历右子树
package algorithm_leetcode; import java.util.ArrayList; import java.util.List; import java.util.Stack; public class Solution_94 { public List<Integer> inorderTraversal(TreeNode root) { // 待处理节点 Stack<TreeNode> stack = new Stack<>(); // 结果 List<Integer> output_arr = new ArrayList<>(); // 如果root为空 if (root == null) { // 返回空的 output_arr return output_arr; } // 初始化为根节点 TreeNode current = root; // 循环 只要当前节点不为空,并且栈不为空 while (current != null || !stack.isEmpty()) { // 当前节点不为空,直到左子树为空 while (current != null) { // 添加到栈 stack.push(current); // 当前节点移动到左子节点 current = current.left; } // 弹出栈节点 current = stack.pop(); // 添加到结果中 output_arr.add(current.val); // 如果有右子节点,就移动到右子节点 current = current.right; } return output_arr; } } class TreeNode { int val; TreeNode left; TreeNode right; TreeNode() {} TreeNode(int val) { this.val = val; } TreeNode(int val, TreeNode left, TreeNode right) { this.val = val; this.left = left; this.right = right; } }