中序遍历二叉树
题目链接:中序遍历二叉树
解法一:
递归遍历比较简单
public List<Integer> inorderTraversal(TreeNode root) {
List<Integer> result = new ArrayList<Integer>();
inorderTraversal(root,result);
return result;
}
public void inorderTraversal(TreeNode root,List<Integer> list) {
if(root==null) {
return;
}
inorderTraversal(root.left,list);
list.add(root.val);
inorderTraversal(root.right,list);
}
解法二:
- 非递归遍历:
- 如果是尾递归一般可以直接修改为循环结构,例如斐波那契数列
- 二叉树不是尾递归,采用手动压栈,弹栈遍历
- 解题过程中遇到的问题有,循环的终止条件是什么?起初我写的是栈结构为空则结束遍历,这里出现了问题,中序遍历时:遍历完左子树就需要弹栈了,如果弹出的为根节点,那么此时栈中已经没有节点,但是还有根节点的右子树没有遍历,所以终止条件为:stack!=空||当前指针指向!=null
- 我采取的是有左子树就压栈当前节点,而不是当前节点不为空就压栈,两者在遍历过程上有小的区别
- 弹栈之后的做法,既然是弹栈说明左子树已经遍历完,那么接下来就是遍历当前弹出的节点,再接着是遍历其右子树,这里不必判断右子树是否为空,直接指针指到右子树就可以了,搭配循环终止条件,为空时刚好可以使其退出循环,不为空继续弹栈
public List<Integer> inorderTraversal(TreeNode root) {
if(root==null) {
return new ArrayList<Integer>();
}
Stack<TreeNode> stack = new Stack<TreeNode>();
List<Integer> result = new ArrayList<Integer>();
TreeNode move = root;
while(!stack.isEmpty()||move!=null) {
if(move!=null) {
if(move.left!=null) {
stack.push(move);
move = move.left;
continue;
}
result.add(move.val);
move = move.right;
}
if(!stack.isEmpty()&&move==null) {
move = stack.pop();
result.add(move.val);
move = move.right;
}
}
return result;
}