二叉树的非递归遍历(前中后)
先序遍历:按照根左右的顺序遍历二叉树;
中序遍历:按照左根右的顺序遍历二叉树;
后序遍历:按照左右根的顺序遍历二叉树。
下图:先序遍历为ABDECF,中序遍历为DBEACF,后序列遍历为DEBFCA.
先序遍历的非递归实现思路:使用栈,入栈时打印节点。
a. 遇到一个节点,访问它,然后把它压栈并打印节点,并去遍历它的左子树;
b. 当左子树遍历结束后,从栈顶弹出该节点并将其指向右儿子,继续a步骤;
c. 当所有节点访问完即最后访问的树节点为空且栈空时,停止。
public void Pre(TreeNode root){
while(!stack.isEmpty() || root !=null ){
if (root !=null){
stack.push(root);
System.out.println(root.getKey());
root = root.getLeft();
}else {
TreeNode node = stack.pop();
root = node.getRight();
}
}
}
中序遍历的非递归思路:使用栈,出栈时打印节点。
a. 遇到节点,访问它,然后把它压栈,并访问它的左子树;
b. 当左子树遍历结束后,从栈顶弹出节点并打印,然后指向其右子树,继续a步骤;
c. 当所有节点访问完且栈空,结束。
public void In(TreeNode root){
while(!stack.isEmpty() || root !=null ){
if (root !=null){
stack.push(root);
root = root.getLeft();
}else {
TreeNode node = stack.pop();
System.out.println(node.getKey());
root = node.getRight();
}
}
}
后序遍历的非递归实现思路:使用栈,出栈时打印,因为后序遍历是要求顺序为左右根,而我们要访问右节点还需要通过根节点,所以我们需要访问两次根节点,第一次访问让其不出栈,只是借助它访问右节点,第二次访问出栈并打印。给节点定义标记位,入栈时标记位设为1,当标记为2的时候出栈并打印。
public void Suc(TreeNode root){
while (!stack.isEmpty() || root != null){
if (root != null){
stack.push(root);
root.setFlags(1);
root = root.getLeft();
}else {
TreeNode node = stack.peek();
if (node.getFlags() ==2){
System.out.println(node.getKey());
stack.pop();
}else {
node.setFlags(2);
root = node.getRight();
}
}
}
}
节点类(get和set以及构造器省略)
public class TreeNode<K extends Comparable<K>> {
private TreeNode parent;
private TreeNode left;
private TreeNode right;
private K key;//可以不用泛型用int定义
private int flags;
}