用栈实现递归(二叉树的遍历)

来源:互联网 发布:关于程序员的搞笑图片 编辑:程序博客网 时间:2024/06/06 02:27

一、先序遍历

public void getNodeByStack(TreeNode node){//用栈实现先序遍历
Stack<TreeNode> stack = new Stack<>();
if(node==null){
return;
}
stack.push(node);
while(!stack.isEmpty()){
TreeNode treeNode = stack.pop();
System.out.print(treeNode.vl+" ");
if(treeNode.right!=null){
stack.push(treeNode.right);
}
if(treeNode.left!=null){
stack.push(treeNode.left);
}
}
}

直接把栈中的top节点拉出来,然后先把有节点放入栈中,再放左节点


二、中序遍历

public void getNodeByStack(TreeNode node){//用栈实现中序遍历
Stack<TreeNode> stack = new Stack<>();
if(node==null){
return;
}
stack.push(node);
while(!stack.isEmpty()){
TreeNode treeNode = stack.pop();
if(treeNode.flag){
if(treeNode.right!=null){
stack.push(treeNode.right);
}
treeNode.flag = false;
stack.push(treeNode);//节点分解完以后要再次进栈
if(treeNode.left!=null){
stack.push(treeNode.left);
}
}else{
System.out.print(treeNode.vl+" ");
}
}
}

中序遍历与先序不同点是要有标志位flag,因为节点的第一次出栈是分解成左右节点,第二次才是遍历输出

0 0
原创粉丝点击