add tests for 334

fishercoder1534 · fishercoder1534 · commit f0489b1f34bd · 2021-10-13T15:21:39.000-07:00
diff --git a/src/main/java/com/fishercoder/solutions/_334.java b/src/main/java/com/fishercoder/solutions/_334.java
@@ -32,15 +32,19 @@ public static class Solution2 {
         /**
          * Time: O(n)
          * Space: O(1)
+         *
+         * Idea: we can keep updating these two numbers:
+         * firstSmallestNumber and secondSmallestNumber
+         * until we found one that's bigger than both
          */
         public boolean increasingTriplet(int[] nums) {
-            int small = Integer.MAX_VALUE;
-            int big = Integer.MAX_VALUE;
+            int firstSmallestNumber = Integer.MAX_VALUE;
+            int secondSmallestNumber = Integer.MAX_VALUE;
             for (int num : nums) {
-                if (num <= small) {
-                    small = num;
-                } else if (num <= big) {
-                    big = num;
+                if (num <= firstSmallestNumber) {
+                    firstSmallestNumber = num;
+                } else if (num <= secondSmallestNumber) {
+                    secondSmallestNumber = num;
                 } else {
                     return true;
                 }
diff --git a/src/test/java/com/fishercoder/_334Test.java b/src/test/java/com/fishercoder/_334Test.java
@@ -0,0 +1,37 @@
+package com.fishercoder;
+
+import com.fishercoder.solutions._334;
+import org.junit.BeforeClass;
+import org.junit.Test;
+
+import static org.junit.Assert.assertEquals;
+
+public class _334Test {
+    private static _334.Solution1 solution1;
+    private static _334.Solution2 solution2;
+
+    @BeforeClass
+    public static void setup() {
+        solution1 = new _334.Solution1();
+        solution2 = new _334.Solution2();
+    }
+
+    @Test
+    public void test1() {
+        assertEquals(true, solution1.increasingTriplet(new int[]{2, 1, 5, 0, 4, 6}));
+        assertEquals(true, solution2.increasingTriplet(new int[]{2, 1, 5, 0, 4, 6}));
+    }
+
+    @Test
+    public void test2() {
+        assertEquals(false, solution1.increasingTriplet(new int[]{5, 4, 3, 2, 1}));
+        assertEquals(false, solution2.increasingTriplet(new int[]{5, 4, 3, 2, 1}));
+    }
+
+    @Test
+    public void test3() {
+        assertEquals(false, solution1.increasingTriplet(new int[]{1, 1, -2, 6}));
+        assertEquals(false, solution2.increasingTriplet(new int[]{1, 1, -2, 6}));
+    }
+
+}
