输入一个树,判断该树是不是合法二分查找树,95题作过生成二分查找树。二分查找树定义以下:java
- 若任意节点的左子树不空,则左子树上全部节点的值均小于它的根节点的值;
- 若任意节点的右子树不空,则右子树上全部节点的值均大于它的根节点的值;
- 任意节点的左、右子树也分别为二叉查找树;
- 没有键值相等的节点。
开始的时候觉得能够很简单的用递归写出来。想法是,左子树是合法二分查找树,右子树是合法二分查找树,而且根节点大于左孩子,小于右孩子,那么当前树就是合法二分查找树。代码以下:node
public boolean isValidBST(TreeNode root) { if (root == null) { return true; } boolean leftVailid = true; boolean rightVaild = true; if (root.left != null) { //大于左孩子而且左子树是合法二分查找树 leftVailid = root.val > root.left.val && isValidBST(root.left); } if (!leftVailid) { return false; } if (root.right != null) { //小于右孩子而且右子树是合法二分查找树 rightVaild = root.val < root.right.val && isValidBST(root.right); } return rightVaild; }
固然,这个解法没有经过。对于下面的解,结果利用上边的解法是错误的。数组
10 / \ 5 15 / \ 6 20
虽然知足左子树是合法二分查找树,右子树是合法二分查找树,而且根节点大于左孩子,小于右孩子,但这个树不是合法的二分查找树。由于右子树中的 6 小于当前根节点 10。因此咱们不该该判断「根节点大于左孩子,小于右孩子」,而是判断「根节点大于左子树中最大的数,小于右子树中最小的数」。spa
public boolean isValidBST(TreeNode root) { if (root == null || root.left == null && root.right == null) { return true; } //左子树是否合法 if (isValidBST(root.left)) { if (root.left != null) { int max = getMaxOfBST(root.left);//获得左子树中最大的数 if (root.val <= max) { //相等的状况,表明有重复的数字 return false; } } } else { return false; } //右子树是否合法 if (isValidBST(root.right)) { if (root.right != null) { int min = getMinOfBST(root.right);//获得右子树中最小的数 if (root.val >= min) { //相等的状况,表明有重复的数字 return false; } } } else { return false; } return true; } private int getMinOfBST(TreeNode root) { int min = root.val; while (root != null) { if (root.val <= min) { min = root.val; } root = root.left; } return min; } private int getMaxOfBST(TreeNode root) { int max = root.val; while (root != null) { if (root.val >= max) { max = root.val; } root = root.right; } return max; }
来利用另外一种思路,参考官方题解。3d
解法一中,咱们是判断根节点是否合法,找到了左子树中最大的数,右子树中最小的数。 由左子树和右子树决定当前根节点是否合法。code
但若是正常的来说,明明先有的根节点,按理说根节点是任何数都行,而不是由左子树和右子树限定。相反,根节点反而决定了左孩子和右孩子的合法取值范围。对象
因此,咱们能够从根节点进行 DFS,而后计算每一个节点应该的取值范围,若是当前节点不符合就返回 false。blog
10 / \ 5 15 / \ / 3 6 7 考虑 10 的范围 10(-inf,+inf) 考虑 5 的范围 10(-inf,+inf) / 5(-inf,10) 考虑 3 的范围 10(-inf,+inf) / 5(-inf,10) / 3(-inf,5) 考虑 6 的范围 10(-inf,+inf) / 5(-inf,10) / \ 3(-inf,5) 6(5,10) 考虑 15 的范围 10(-inf,+inf) / \ 5(-inf,10) 15(10,+inf) / \ 3(-inf,5) 6(5,10) 考虑 7 的范围,出现不符合返回 false 10(-inf,+inf) / \ 5(-inf,10) 15(10,+inf) / \ / 3(-inf,5) 6(5,10) 7(10,15)
能够观察到,左孩子的范围是 (父结点左边界,父节点的值),右孩子的范围是(父节点的值,父节点的右边界)。递归
还有个问题,java 里边没有提供负无穷和正无穷,用什么数来表示呢?队列
方案一,假设咱们的题目的数值都是 Integer 范围的,那么咱们用不在 Integer 范围的数字来表示负无穷和正无穷。用 long 去存储。
public boolean isValidBST(TreeNode root) { long maxValue = (long)Integer.MAX_VALUE + 1; long minValue = (long)Integer.MIN_VALUE - 1; return getAns(root, minValue, maxValue); } private boolean getAns(TreeNode node, long minVal, long maxVal) { if (node == null) { return true; } if (node.val <= minVal) { return false; } if (node.val >= maxVal) { return false; } return getAns(node.left, minVal, node.val) && getAns(node.right, node.val, maxVal); }
方案二:传入 Integer 对象,而后 null 表示负无穷和正无穷。而后利用 JAVA 的自动装箱拆箱,数值的比较能够直接用不等号。
public boolean isValidBST(TreeNode root) { return getAns(root, null, null); } private boolean getAns(TreeNode node, Integer minValue, Integer maxValue) { if (node == null) { return true; } if (minValue != null && node.val <= minValue) { return false; } if (maxValue != null && node.val >= maxValue) { return false; } return getAns(node.left, minValue, node.val) && getAns(node.right, node.val, maxValue); }
解法二其实就是树的 DFS,也就是二叉树的先序遍历,而后在遍历过程当中,判断当前的值是是否在区间中。因此咱们能够用栈来模拟递归过程。
public boolean isValidBST(TreeNode root) { if (root == null || root.left == null && root.right == null) { return true; } //利用三个栈来保存对应的节点和区间 LinkedList<TreeNode> stack = new LinkedList<>(); LinkedList<Integer> minValues = new LinkedList<>(); LinkedList<Integer> maxValues = new LinkedList<>(); //头结点入栈 TreeNode pNode = root; stack.push(pNode); minValues.push(null); maxValues.push(null); while (pNode != null || !stack.isEmpty()) { if (pNode != null) { //判断栈顶元素是否符合 Integer minValue = minValues.peek(); Integer maxValue = maxValues.peek(); TreeNode node = stack.peek(); if (minValue != null && node.val <= minValue) { return false; } if (maxValue != null && node.val >= maxValue) { return false; } //将左孩子加入到栈 if(pNode.left!=null){ stack.push(pNode.left); minValues.push(minValue); maxValues.push(pNode.val); } pNode = pNode.left; } else { // pNode == null && !stack.isEmpty() //出栈,将右孩子加入栈中 TreeNode node = stack.pop(); minValues.pop(); Integer maxValue = maxValues.pop(); if(node.right!=null){ stack.push(node.right); minValues.push(node.val); maxValues.push(maxValue); } pNode = node.right; } } return true; }
上边的 DFS 能够看出来一个缺点,就是咱们判断完当前元素后并无出栈,后续还会回来获得右孩子后才会出栈。因此其实咱们能够用 BFS,利用一个队列,一层一层的遍历,遍历完一个就删除一个。
public boolean isValidBST(TreeNode root) { if (root == null || root.left == null && root.right == null) { return true; } //利用三个队列来保存对应的节点和区间 Queue<TreeNode> queue = new LinkedList<>(); Queue<Integer> minValues = new LinkedList<>(); Queue<Integer> maxValues = new LinkedList<>(); //头结点入队列 TreeNode pNode = root; queue.offer(pNode); minValues.offer(null); maxValues.offer(null); while (!queue.isEmpty()) { //判断队列的头元素是否符合条件而且出队列 Integer minValue = minValues.poll(); Integer maxValue = maxValues.poll(); pNode = queue.poll(); if (minValue != null && pNode.val <= minValue) { return false; } if (maxValue != null && pNode.val >= maxValue) { return false; } //左孩子入队列 if(pNode.left!=null){ queue.offer(pNode.left); minValues.offer(minValue); maxValues.offer(pNode.val); } //右孩子入队列 if(pNode.right!=null){ queue.offer(pNode.right); minValues.offer(pNode.val); maxValues.offer(maxValue); } } return true; }
参考这里。
解法三中咱们用了先序遍历 和 BFS,如今来考虑中序遍历。中序遍历在 94 题中已经考虑过了。那么中序遍历在这里有什么好处呢?
中序遍历顺序会是左孩子,根节点,右孩子。二分查找树的性质,左孩子小于根节点,根节点小于右孩子。
是的,若是咱们将中序遍历的结果输出,那么将会到的一个从小到大排列的序列。
因此咱们只须要进行一次中序遍历,将遍历结果保存,而后判断该数组是不是从小到大排列的便可。
更近一步,因为咱们只须要临近的两个数的相对关系,因此咱们只须要在遍历过程当中,把当前遍历的结果和上一个结果比较便可。
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; }
这几天都是二叉树的相关题,主要是对前序遍历,中序遍历的理解,以及 DFS,若是再用好递归,利用栈模拟递归,题目就很好解了。
更多详细通俗题解详见 leetcode.wang 。