题目
给定一个二叉树的根节点root,返回它的中序遍历结果。
数据范围:树上节点数满足0≤n≤1000,树上每个节点的值满足 −1000≤val≤1000。
进阶:空间复杂度O(n),时间复杂度O(n)。
示例1
输入:{1,2,#,#,3}
返回值:[2,3,1]
说明:
示例2
输入:{}
返回值:[]
示例3
输入:{1,2}
返回值:[2,1]
说明:
示例4
输入:{1,#,2}
返回值:[1,2]
说明:
备注:树中节点数目在范围 [0, 100] 内;树中的节点的值在[-100,100]以内
思路1:递归(左根右)
代码1
import java.util.*;
public class Solution {
public int[] inorderTraversal (TreeNode root) {
//添加遍历结果的数组
List<Integer> list = new ArrayList();
//递归中序遍历
inorder(list, root);
//返回的结果
int[] res = new int[list.size()];
for(int i = 0; i < list.size(); i++){
res[i] = list.get(i);
}
return res;
}
public void inorder(List<Integer> list, TreeNode root){
//遇到空节点则返回
if(root == null){
return;
}
//左
inorder(list, root.left);
//根
list.add(root.val);
//右
inorder(list, root.right);
}
}
思路2:迭代,栈
第二次访问根节点(走完左树),根节点才能真正遍历输出。
代码2
import java.util.*;
public class Solution {
public int[] inorderTraversal (TreeNode root) {
//添加遍历结果的数组
List<Integer> list = new ArrayList();
Stack<TreeNode> s = new Stack<TreeNode>();
//空树返回空数组
if(root == null){
return new int[0];
}
//当树节点不为空或栈中有节点时
while(root != null || !s.isEmpty()){
//一路向左走一直走到最左侧的最后一个节点
while(root != null){
s.push(root);
root = root.left;
}
//当前左树已经全部入栈,访问该节点
TreeNode node = s.pop();
//此时访问根节点
list.add(node.val);
//继续访问右子树
root = node.right;
}
//返回的结果
int[] res = new int[list.size()];
for(int i = 0; i < list.size(); i++){
res[i] = list.get(i);
}
return res;
}
}