一、先序遍历
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,因为节点的第一次出栈是分解成左右节点,第二次才是遍历输出