refactor 495

fishercoder1534 · fishercoder1534 · commit ff59d5e7a455 · 2019-05-16T07:13:21.000-07:00
diff --git a/src/main/java/com/fishercoder/solutions/_495.java b/src/main/java/com/fishercoder/solutions/_495.java
@@ -32,20 +32,22 @@
  */
 public class _495 {
 
-    public int findPoisonedDuration(int[] timeSeries, int duration) {
-        if (timeSeries == null || timeSeries.length == 0 || duration == 0) {
-            return 0;
-        }
-        int totalDuration = 0;
-        for (int i = 0; i < timeSeries.length - 1; i++) {
-            if ((timeSeries[i + 1] - timeSeries[i]) >= duration) {
-                totalDuration += duration;
-            } else {
-                totalDuration += (timeSeries[i + 1] - timeSeries[i]);
+    public static class Solution1 {
+        public int findPoisonedDuration(int[] timeSeries, int duration) {
+            if (timeSeries == null || timeSeries.length == 0 || duration == 0) {
+                return 0;
+            }
+            int totalDuration = 0;
+            for (int i = 0; i < timeSeries.length - 1; i++) {
+                if ((timeSeries[i + 1] - timeSeries[i]) >= duration) {
+                    totalDuration += duration;
+                } else {
+                    totalDuration += (timeSeries[i + 1] - timeSeries[i]);
+                }
             }
+            totalDuration += duration;//plus the last one duration
+            return totalDuration;
         }
-        totalDuration += duration;//plus the last one duration
-        return totalDuration;
     }
 
 }
diff --git a/src/test/java/com/fishercoder/_495Test.java b/src/test/java/com/fishercoder/_495Test.java
@@ -10,6 +10,7 @@
  * Created by fishercoder on 5/8/17.
  */
 public class _495Test {
+    _495.Solution1 solution1 = new _495.Solution1();
     private static int actual = 0;
     private static int expected = 0;
     private static int[] timeSeries;
@@ -25,30 +26,27 @@ public void setup() {
 
     @Test
     public void test1() {
-        _495 test = new _495();
-        timeSeries = new int[]{1,4};
+        timeSeries = new int[]{1, 4};
         duration = 2;
-        actual = test.findPoisonedDuration(timeSeries, duration);
+        actual = solution1.findPoisonedDuration(timeSeries, duration);
         expected = 4;
         assertEquals(expected, actual);
     }
 
     @Test
     public void test2() {
-        _495 test = new _495();
-        timeSeries = new int[]{1,2};
+        timeSeries = new int[]{1, 2};
         duration = 2;
-        actual = test.findPoisonedDuration(timeSeries, duration);
+        actual = solution1.findPoisonedDuration(timeSeries, duration);
         expected = 3;
         assertEquals(expected, actual);
     }
 
     @Test
     public void test3() {
-        _495 test = new _495();
         timeSeries = new int[]{};
         duration = 100000;
-        actual = test.findPoisonedDuration(timeSeries, duration);
+        actual = solution1.findPoisonedDuration(timeSeries, duration);
         expected = 0;
         assertEquals(expected, actual);
     }
