add tests for 98

fishercoder1534 · fishercoder1534 · commit 3493dae07b72 · 2017-05-17T07:18:29.000-07:00
diff --git a/src/main/java/com/stevesun/solutions/_98.java b/src/main/java/com/stevesun/solutions/_98.java
@@ -24,6 +24,21 @@
  Binary tree [1,2,3], return false.
  */
 public class _98 {
+    class MoreConciseSolution {
+
+        public boolean isValidBST(TreeNode root) {
+            return valid(root, null, null);
+        }
+
+        private boolean valid(TreeNode root, Integer min, Integer max) {
+            if (root == null) return true;
+            if ((min != null && root.val <= min)
+                    || (max != null && root.val >= max)) return false;
+            return valid(root.left, min, root.val) && valid(root.right, root.val, max);
+        }
+    }
+
+
     public boolean isValidBST(TreeNode root) {
         if (root == null) return true;
         return dfs(root.left, Long.MIN_VALUE, root.val) && dfs(root.right, root.val, Long.MAX_VALUE);
diff --git a/src/test/java/com/stevesun/_98Test.java b/src/test/java/com/stevesun/_98Test.java
@@ -0,0 +1,43 @@
+package com.stevesun;
+
+import com.stevesun.common.classes.TreeNode;
+import com.stevesun.solutions._48;
+import com.stevesun.solutions._98;
+import org.junit.BeforeClass;
+import org.junit.Test;
+
+import static org.junit.Assert.assertEquals;
+
+/**
+ * Created by stevesun on 5/17/17.
+ */
+public class _98Test {
+    private static _98 test;
+    private static TreeNode root;
+
+    @BeforeClass
+    public static void setup(){
+        test = new _98();
+    }
+
+    @Test
+    public void test1(){
+        root = new TreeNode(2);
+        root.left = new TreeNode(1);
+        root.right = new TreeNode(3);
+        assertEquals(true, test.isValidBST(root));
+    }
+
+    @Test
+    public void test2(){
+        root = new TreeNode(0);
+        assertEquals(true, test.isValidBST(root));
+    }
+
+    @Test
+    public void test3(){
+        root = new TreeNode(1);
+        root.left = new TreeNode(1);
+        assertEquals(false, test.isValidBST(root));
+    }
+}
