[LeetCode] 98. Validate Binary Search Tree
Given the root of a binary tree, determine if it is a valid binary search tree (BST).
A valid BST is defined as follows:
- The left subtree of a node contains only nodes with keys less than the node's key.
- The right subtree of a node contains only nodes with keys greater than the node's key.
- Both the left and right subtrees must also be binary search trees.
Example1:
Input: root = [2,1,3]
Output: true
Example2:
Input: root = [5,1,4,null,null,3,6]
Output: false
Explanation: The root node's value is 5 but its right child's value is 4.
Constraints:
- The number of nodes in the tree is in the range [1, 104].
- -231 <= Node.val <= 231 - 1
所谓的搜索二叉树,就是一颗二叉树上每个子树的左子树的值都小于它,它的右子树都数值都大于它。根据这个定义,我们可以想到两个方法。
方法一:
二叉树中序遍历(左根右), 得到一个数组,这个数组单调递增,那么根据定义,就一定是搜索二叉树。
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public boolean isValidBST(TreeNode root) {
if (root == null) {
return true;
}
List rst = new ArrayList<>();
helper(root, rst);
return isIncrease(rst);
}
private void helper(TreeNode root, List rst) {
if (root == null) {
return;
}
helper(root.left, rst);
rst.add(root.val);
helper(root.right, rst);
}
private boolean isIncrease(List rst) {
for (int i = 0; i < rst.size() - 1; i++) {
if (rst.get(i) >= rst.get(i + 1)) {
return false;
}
}
return true;
}
}
第二个思路,就是用基本的递归。但是因为需要判断每层左子树是不是都小于根,右子树是不是都大于根,所以我们需要三个信息。
1)是不是平衡 => 左边的子树是不是平衡+右边子树是不是平衡+该层是不是平衡
2)左子树的最大值 => 只要最大值小于根就所有都小于
3)右子树的最小值 => 只要最小值大于根就所有都大于
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
public class Info {
int leftMax;
int rightMin;
boolean isValid;
Info(int leftMax, int rightMin, boolean isValid) {
this.leftMax = leftMax;
this.rightMin = rightMin;
this.isValid = isValid;
}
}
class Solution {
public boolean isValidBST(TreeNode root) {
if (root == null) {
return true;
}
Info rst = helper(root);
return rst.isValid;
}
private Info helper(TreeNode root) {
if (root == null) {
return null;
}
if (root.left == null && root.right == null) {
return new Info(root.val, root.val, true);
}
Info l = helper(root.left);
Info r = helper(root.right);
int leftMax = root.val;
int rightMin = root.val;
boolean isValid = true;
if ( l != null) {
isValid = l.leftMax < root.val && l.isValid && isValid;
leftMax = Math.max(l.leftMax, leftMax);
rightMin = Math.min(l.rightMin, rightMin);
}
if (r != null) {
isValid = r.rightMin > root.val && r.isValid && isValid;
rightMin = Math.min(r.rightMin, rightMin);
leftMax = Math.max(r.leftMax, leftMax);
}
return new Info(leftMax, rightMin, isValid);
}
}