目录
二叉树遍历的分类
前序遍历: 先输出父节点,再遍历左子树和右子树
中序遍历: 先遍历左子树,再输出父节点,再遍历右子树
后序遍历: 先遍历左子树,再遍历右子树,最后输出父节点
(看输出父节点的顺序,就确定是前序,中序还是后序)
1.前序遍历
过程:
前序遍历首先访问根结点然后遍历左子树,最后遍历右子树。在遍历左、右子树时,仍然先访问根结点,然后遍历左子树,最后遍历右子树。
代码实现:
//非递归
public void Tree1(TreeNode root) {
Stack<TreeNode> stack = new Stack<>();//用一个栈来存放树中的节点
while(root != null || !stack.isEmpty()) {//只要当前节点不为空(即当前节点的左右子树没有访问完毕)或者栈中还有节点(还有节点没有访问)
while (root != null) {//一直往左走
stack.push(root);//根节点入栈
System.out.println(root.val);//打印根节点
root = root.left;//访问左子树
}
root = stack.pop();//取出根节点
root = root.right;//访问右子树
}
}
2.中序遍历
过程:
中序遍历首先遍历左子树,然后访问根结点,最后遍历右子树。
代码实现:
//非递归
public class Tree2 {
public void inorderTraversal(TreeNode root){
Stack<TreeNode> stack = new Stack<>();
while(root != null || !stack.isEmpty()){//只要当前节点不为空(即当前节点的左右子树没有访问完毕)或者栈中还有节点(还有节点没有访问)
while(root != null){
stack.push(root);//根节点入栈
root = root.left;//访问左子树
}
root = stack.pop();//取出左子树的根节点
System.out.println(root.val);//打印根节点
root = root.right;//访问右子树
}
}
}
3. 后序遍历
过程:
首先遍历左子树,然后遍历右子树,最后遍历访问根结点,在遍历左、右子树时,仍然先遍历左子树,然后遍历右子树,最后遍历根结点。
代码实现:
//非递归
public class PostOrderTraversal {
public void postorderTraversal(TreeNode root) {
Stack<TreeNode> stack = new Stack<>();
TreeNode node = root;
TreeNode prev = null;
while(node != null || !stack.isEmpty()){
while(node != null){
//访问左子树
stack.push(node);
node = node.left;
}
//判断栈顶元素(根)
node = stack.peek();
//1.如果此时的根的右子树为空node.right == null
//2.如果此时的根的右子树已经访问过了node.right == prev(prev记录的是上次访问打印的节点)
if(node.right == null || node.right == prev){
//打印根节点,并出栈,将打印过的节点从栈中删除
stack.pop();
System.out.println(node.val);
//记录prev,表示以当前prev为根的子树已经访问过了
prev = node;
//node置null就不会再次访问以node为根节点的左右子树,这里的node既然已经打印,说明它的左右子树早已访问完毕
node = null;
}else{
//访问右子树
node = node.right;
}
}
}
}
(注意:在遍历前,都要去判断树是否为空)