我们经常会使用递归的方法去遍历一棵二叉树,这个实现起来也非常简单,但要使用非递归去遍历二叉树还是需要一些编程技巧的。我们很容易想到了使用栈去处理,下面是具体实现
import java.util.Stack;
public class Test {
public static void main(String[] args) {
TreeNode root = new TreeNode(1);
root.left = new TreeNode(2);
root.right = new TreeNode(3);
root.left.left = new TreeNode(4);
root.left.right = new TreeNode(5);
root.right.left = new TreeNode(6);
root.right.right = new TreeNode(7);
midToPrintBinaryTreeByStack(root);
}
public static void midToPrintBinaryTreeByStack(TreeNode root){
Stack<TreeNode> stack = new Stack<>();
if(root==null)
return;
stack.push(root);
TreeNode node;
while(!stack.isEmpty()){
node = stack.peek();
//如果有左孩子,那么将左孩子入栈,且将左指针赋为空
if(node.left!=null){
stack.push(node.left);
node.left = null;
}else if(node.right!=null){
//如果有右孩子,那么打印当前的节点(它是中)并且弹出栈顶,将右孩子入栈
System.out.print(node.value+" ");
stack.pop();
stack.push(node.right);
}else{
//如果当前为叶子节点,直接打印并且弹出栈顶
System.out.print(node.value+" ");
stack.pop();
}
}
}
static class TreeNode{
int value;
public TreeNode(int value){
this.value = value;
}
TreeNode left;
TreeNode right;
}
}
结果
4 2 5 1 6 3 7
这样实现有一个缺点就是会破坏树的结构,因为将所有节点的左指针设置成了null