refactor 527

fishercoder1534 · fishercoder1534 · commit 0ba2286057cd · 2019-06-12T09:46:38.000-07:00
diff --git a/src/main/java/com/fishercoder/solutions/_527.java b/src/main/java/com/fishercoder/solutions/_527.java
@@ -28,48 +28,48 @@
  */
 public class _527 {
 
-    /**reference: https://discuss.leetcode.com/topic/82613/really-simple-and-straightforward-java-solution*/
-    public List<String> wordsAbbreviation(List<String> dict) {
-        int len = dict.size();
-        String[] ans = new String[len];
-        int[] prefix = new int[len];
-        for (int i = 0; i < len; i++) {
-            prefix[i] = 1;
-            ans[i] = abbreviate(dict.get(i), 1); // make abbreviation for each string
-        }
-        for (int i = 0; i < len; i++) {
-            while (true) {
-                HashSet<Integer> set = new HashSet<>();
-                for (int j = i + 1; j < len; j++) {
-                    if (ans[j].equals(ans[i])) {
-                        set.add(j); // check all strings with the same abbreviation
+    public static class Solution1 {
+        /**
+         * reference: https://discuss.leetcode.com/topic/82613/really-simple-and-straightforward-java-solution
+         */
+        public List<String> wordsAbbreviation(List<String> dict) {
+            int len = dict.size();
+            String[] ans = new String[len];
+            int[] prefix = new int[len];
+            for (int i = 0; i < len; i++) {
+                prefix[i] = 1;
+                ans[i] = abbreviate(dict.get(i), 1); // make abbreviation for each string
+            }
+            for (int i = 0; i < len; i++) {
+                while (true) {
+                    HashSet<Integer> set = new HashSet<>();
+                    for (int j = i + 1; j < len; j++) {
+                        if (ans[j].equals(ans[i])) {
+                            set.add(j); // check all strings with the same abbreviation
+                        }
+                    }
+                    if (set.isEmpty()) {
+                        break;
+                    }
+                    set.add(i);
+                    for (int k : set) {
+                        ans[k] = abbreviate(dict.get(k), ++prefix[k]); // increase the prefix
                     }
-                }
-                if (set.isEmpty()) {
-                    break;
-                }
-                set.add(i);
-                for (int k : set) {
-                    ans[k] = abbreviate(dict.get(k), ++prefix[k]); // increase the prefix
                 }
             }
+            return Arrays.asList(ans);
         }
-        return Arrays.asList(ans);
-    }
 
-    private String abbreviate(String word, int k) {
-        if (k + 2 >= word.length()) {
-            return word;
+        private String abbreviate(String word, int k) {
+            if (k + 2 >= word.length()) {
+                return word;
+            }
+            StringBuilder stringBuilder = new StringBuilder();
+            stringBuilder.append(word.substring(0, k));
+            stringBuilder.append(word.length() - 1 - k);
+            stringBuilder.append(word.substring(word.length() - 1));
+            return stringBuilder.toString();
         }
-        StringBuilder stringBuilder = new StringBuilder();
-        stringBuilder.append(word.substring(0, k));
-        stringBuilder.append(word.length() - 1 - k);
-        stringBuilder.append(word.substring(word.length() - 1));
-        return stringBuilder.toString();
     }
 
-    public static void main(String... args) {
-        _527 test = new _527();
-        System.out.println(test.abbreviate("saaap", 2));
-    }
 }
diff --git a/src/test/java/com/fishercoder/_527Test.java b/src/test/java/com/fishercoder/_527Test.java
@@ -0,0 +1,24 @@
+package com.fishercoder;
+
+import com.fishercoder.solutions._527;
+import org.junit.BeforeClass;
+import org.junit.Test;
+
+import java.util.Arrays;
+
+import static org.junit.Assert.assertEquals;
+
+public class _527Test {
+    private static _527.Solution1 solution1;
+
+    @BeforeClass
+    public static void setup() {
+        solution1 = new _527.Solution1();
+    }
+
+    @Test
+    public void test1() {
+        assertEquals(Arrays.asList("l2e", "god", "internal", "me", "i6t", "interval", "inte4n", "f2e", "intr4n"), solution1.wordsAbbreviation(Arrays.asList("like", "god", "internal", "me", "internet", "interval", "intension", "face", "intrusion")));
+    }
+
+}
