验证 BST

Question

关于通过顺序遍历验证 BST 的一个简单问题

public boolean isValidBST(TreeNode root) {
   if (root == null) return true;
   Stack<TreeNode> stack = new Stack<>();
   TreeNode pre = null;
   while (root != null || !stack.isEmpty()) {
      while (root != null) {
         stack.push(root);
         root = root.left;
      }
      root = stack.pop();
      if(pre != null && root.val <= pre.val) return false;
      pre = root;
      root = root.right;
   }
   return true;
}

冒着听起来很傻的风险，倒数第二行 root = root.right 的意义何在，因为它不会在 while 循环的下一次迭代之后立即重新分配给 root = stack.pop()？

Answer 1

倒数第二行 root = root.right 的意义是什么

如果你不访问节点的右子节点，你实际上是在一次又一次地处理同一个节点..等等，你总是会陷入无限循环。

在 while 循环的下一次迭代之后，它不会立即重新分配给 root = stack.pop() 吗？

没有。

 while (root != null) {
     stack.push(root);
     root = root.left;
  }
  root = stack.pop();

到目前为止，您位于树的最左侧节点。在您遍历当前节点的右子节点之前，它将始终指向同一个节点。

请看以下内容：

        4
       / 
      3 
     / 
    2
   / \
   1  0

在内部 while 循环之后，您的堆栈将如下所示：4->3->2->1 除非您分别访问 1 和 2 的右子节点，否则它将始终在 2 和 1 节点之间循环。

因此，

root = root.right;

终止循环并遍历树的所有节点非常重要。

来自印度的晚安！