//中序遍历递归
public List<Integer> inorderTraversal_1(TreeNode root) {
Stack<TreeNode> stack = new Stack<>();
LinkedList<Integer> ans = new LinkedList<>();
subInorderTraversal(root, ans);
return ans;
}
private void subInorderTraversal(TreeNode root, List<Integer> list){
if(root == null){
return;
}
subInorderTraversal(root.left, list);
list.add(root.val);
subInorderTraversal(root.right, list);
}
//中序遍历非递归,判断条件为当前节点和栈至少有一个不为空:
//当cur不为空时,其实表明cur还有left节点(如果cur是叶子,left就是null,在下一循环cur就是null了),那么就一直将cur入栈,并再次访问cur的left;
//直到cur为空,表明此时已经访问到最左节点了,此时弹出stack顶部,加入ans队列,并将顶部的right赋值给cur,开始下一次循环(这样下一次就会循环入栈right的所有left);
//直到都为空,树遍历完成。
public List<Integer> inorderTraversal_2(TreeNode root) {
Stack<TreeNode> stack = new Stack<>();
LinkedList<Integer> ans = new LinkedList<>();
TreeNode cur = root;
while(cur != null || !stack.empty()){
if(cur != null){
stack.push(cur);
cur = cur.left;
}else{
cur = stack.pop();
ans.add(cur.val);
cur = cur.right;
}
}
return ans;
}
网友评论