refactor 352

fishercoder1534 · fishercoder1534 · commit 92f32d6ba543 · 2019-01-22T06:40:48.000-08:00
diff --git a/src/main/java/com/fishercoder/solutions/_352.java b/src/main/java/com/fishercoder/solutions/_352.java
@@ -24,41 +24,46 @@
  */
 public class _352 {
 
-    public static class SummaryRanges {
+    public static class Solution1 {
+        public static class SummaryRanges {
 
-        /**
-         * Use treemap, key is the start of the interval.
-         */
-        TreeMap<Integer, Interval> treeMap;
+            /**
+             * Use treemap, key is the start of the interval.
+             */
+            TreeMap<Integer, Interval> treeMap;
 
-        /**
-         * Initialize your data structure here.
-         */
-        public SummaryRanges() {
-            treeMap = new TreeMap<>();
-        }
-
-        public void addNum(int val) {
-            if (treeMap.containsKey(val)) {
-                return;
+            /**
+             * Initialize your data structure here.
+             */
+            public SummaryRanges() {
+                treeMap = new TreeMap<>();
             }
-            Integer lower = treeMap.lowerKey(val);
-            Integer higher = treeMap.higherKey(val);
-            if (lower != null && higher != null && treeMap.get(lower).end + 1 == val && higher == val + 1) {
-                treeMap.get(lower).end = treeMap.get(higher).end;
-                treeMap.remove(higher);
-            } else if (lower != null && treeMap.get(lower).end + 1 >= val) {
-                treeMap.get(lower).end = Math.max(treeMap.get(lower).end, val);
-            } else if (higher != null && higher == val + 1) {
-                treeMap.put(val, new Interval(val, treeMap.get(higher).end));
-                treeMap.remove(higher);
-            } else {
-                treeMap.put(val, new Interval(val, val));
+
+            public void addNum(int val) {
+                if (treeMap.containsKey(val)) {
+                    return;
+                }
+                Integer lower = treeMap.lowerKey(val);
+                Integer higher = treeMap.higherKey(val);
+                if (lower != null
+                    && higher != null
+                    && treeMap.get(lower).end + 1 == val
+                    && higher == val + 1) {
+                    treeMap.get(lower).end = treeMap.get(higher).end;
+                    treeMap.remove(higher);
+                } else if (lower != null && treeMap.get(lower).end + 1 >= val) {
+                    treeMap.get(lower).end = Math.max(treeMap.get(lower).end, val);
+                } else if (higher != null && higher == val + 1) {
+                    treeMap.put(val, new Interval(val, treeMap.get(higher).end));
+                    treeMap.remove(higher);
+                } else {
+                    treeMap.put(val, new Interval(val, val));
+                }
             }
-        }
 
-        public List<Interval> getIntervals() {
-            return new ArrayList<>(treeMap.values());
+            public List<Interval> getIntervals() {
+                return new ArrayList<>(treeMap.values());
+            }
         }
     }
 }
diff --git a/src/test/java/com/fishercoder/_352Test.java b/src/test/java/com/fishercoder/_352Test.java
@@ -8,38 +8,35 @@
 
 import java.util.List;
 
-/**
- * Created by stevesun on 6/1/17.
- */
 public class _352Test {
-    private static _352.SummaryRanges test;
-    private static List<Interval> actual;
-
-    @BeforeClass
-    public static void setup() {
-        test = new _352.SummaryRanges();
-    }
-
-    @Test
-    public void test1() {
-        test.addNum(1);
-        actual = test.getIntervals();
-        CommonUtils.printIntervals(actual);
-
-        test.addNum(3);
-        actual = test.getIntervals();
-        CommonUtils.printIntervals(actual);
-
-        test.addNum(7);
-        actual = test.getIntervals();
-        CommonUtils.printIntervals(actual);
-
-        test.addNum(2);
-        actual = test.getIntervals();
-        CommonUtils.printIntervals(actual);
-
-        test.addNum(6);
-        actual = test.getIntervals();
-        CommonUtils.printIntervals(actual);
-    }
+  private static _352.Solution1.SummaryRanges test;
+  private static List<Interval> actual;
+
+  @BeforeClass
+  public static void setup() {
+    test = new _352.Solution1.SummaryRanges();
+  }
+
+  @Test
+  public void test1() {
+    test.addNum(1);
+    actual = test.getIntervals();
+    CommonUtils.printIntervals(actual);
+
+    test.addNum(3);
+    actual = test.getIntervals();
+    CommonUtils.printIntervals(actual);
+
+    test.addNum(7);
+    actual = test.getIntervals();
+    CommonUtils.printIntervals(actual);
+
+    test.addNum(2);
+    actual = test.getIntervals();
+    CommonUtils.printIntervals(actual);
+
+    test.addNum(6);
+    actual = test.getIntervals();
+    CommonUtils.printIntervals(actual);
+  }
 }
