二叉树的单栈后序遍历

Question

我想仅使用一个堆栈对二叉树进行后序遍历。这是我的代码，首先我将左侧元素推入堆栈，直到达到null。然后我弹出一个元素并检查弹出的元素和当前堆栈顶部的右侧元素是否相同。但不知何故，这将进入无限循环。你能告诉我为什么吗？？

public void postorderonestack(BinNode root)
{
    BinStack s=new BinStack();

    while(true)
    {
        if(root!=null)
        {
           s.push(root);
           root=root.getLeft();
        }
        else
        {
            if(s.isEmpty())
            {
                System.out.println("Stack is Empty");
                return;
            }

            else if( s.top().getRight()==null)
            {
                root=s.pop();
                System.out.println(root.getKey());

                if(root==s.top().getRight())
                {
                   System.out.print(s.top().getKey());
                   s.pop();
                }
            }

            if(!s.isEmpty())
              root=s.top().getRight();

           else root=null;
        }
    }
}

Answer 1

这段代码中有一个无限循环，让我们考虑一棵右偏树来说明这一点：

1有一个合适的孩子2

2有一个合适的孩子3

3是叶节点

在前3个循环中，1、2、3将被推到堆栈上。现在在第四个循环中，它进入了其他部分，现在它打印出3，然后是2，两个都弹出。

声明之后

else if( s.top().getRight()==null) 

声明

if(!s.isEmpty()) 

将再次推动堆栈上的2。这会导致无限循环。

代码有缺陷。

声明

if(root==s.top().getRight()) 

应该变成一个while循环来解决这个问题。

Answer 2

为什么不递归地做呢？

public void postorder(BinNode root) {
    if (root == null)
        return;
    postorder(root.getLeft());
    postorder(root.getRight());
    System.out.println(root.getKey());
}

Answer 3

这是我对另一个问题（无递归二叉树的后序遍历）的回答中的代码。它只使用一个堆栈，不使用访问标志。

private void postorder(Node head) {
  if (head == null) {
    return;
  }
  LinkedList<Node> stack = new LinkedList<Node>();
  stack.push(head);

  while (!stack.isEmpty()) {
    Node next = stack.peek();

    if (next.right == head || next.left == head ||
        (next.left == null && next.right == null)) {
      stack.pop();
      System.out.println(next.value);
      head = next;
    }
    else {
      if (next.right != null) {
        stack.push(next.right);
      }
      if (next.left != null) {
        stack.push(next.left);
      }
    }
  }
}