refactor 647

Author: fishercoder1534

src/main/java/com/fishercoder/solutions/_648.java

@@ -29,62 +29,64 @@
 
  */
 public class _648 {
-    public String replaceWords(List<String> dict, String sentence) {
-        String[] tokens = sentence.split(" ");
-        TrieNode trie = buildTrie(dict);
-        return replaceWords(tokens, trie);
-    }
+    public static class Solution1 {
+        public String replaceWords(List<String> dict, String sentence) {
+            String[] tokens = sentence.split(" ");
+            TrieNode trie = buildTrie(dict);
+            return replaceWords(tokens, trie);
+        }
 
-    private String replaceWords(String[] tokens, TrieNode root) {
-        StringBuilder stringBuilder = new StringBuilder();
-        for (String token : tokens) {
-            stringBuilder.append(getShortestReplacement(token, root));
-            stringBuilder.append(" ");
+        private String replaceWords(String[] tokens, TrieNode root) {
+            StringBuilder stringBuilder = new StringBuilder();
+            for (String token : tokens) {
+                stringBuilder.append(getShortestReplacement(token, root));
+                stringBuilder.append(" ");
+            }
+            return stringBuilder.substring(0, stringBuilder.length() - 1);
         }
-        return stringBuilder.substring(0, stringBuilder.length() - 1);
-    }
 
-    private String getShortestReplacement(String token, final TrieNode root) {
-        TrieNode temp = root;
-        StringBuilder stringBuilder = new StringBuilder();
-        for (char c : token.toCharArray()) {
-            stringBuilder.append(c);
-            if (temp.children[c - 'a'] != null) {
-                if (temp.children[c - 'a'].isWord) {
-                    return stringBuilder.toString();
+        private String getShortestReplacement(String token, final TrieNode root) {
+            TrieNode temp = root;
+            StringBuilder stringBuilder = new StringBuilder();
+            for (char c : token.toCharArray()) {
+                stringBuilder.append(c);
+                if (temp.children[c - 'a'] != null) {
+                    if (temp.children[c - 'a'].isWord) {
+                        return stringBuilder.toString();
+                    }
+                    temp = temp.children[c - 'a'];
+                } else {
+                    return token;
                 }
-                temp = temp.children[c - 'a'];
-            } else {
-                return token;
             }
+            return token;
         }
-        return token;
-    }
 
-    private TrieNode buildTrie(List<String> dict) {
-        TrieNode root = new TrieNode(' ');
-        for (String word : dict) {
-            TrieNode temp = root;
-            for (char c : word.toCharArray()) {
-                if (temp.children[c - 'a'] == null) {
-                    temp.children[c - 'a'] = new TrieNode(c);
+        private TrieNode buildTrie(List<String> dict) {
+            TrieNode root = new TrieNode(' ');
+            for (String word : dict) {
+                TrieNode temp = root;
+                for (char c : word.toCharArray()) {
+                    if (temp.children[c - 'a'] == null) {
+                        temp.children[c - 'a'] = new TrieNode(c);
+                    }
+                    temp = temp.children[c - 'a'];
                 }
-                temp = temp.children[c - 'a'];
+                temp.isWord = true;
             }
-            temp.isWord = true;
+            return root;
         }
-        return root;
-    }
 
-    public class TrieNode {
-        char val;
-        TrieNode[] children;
-        boolean isWord;
+        public class TrieNode {
+            char val;
+            TrieNode[] children;
+            boolean isWord;
 
-        public TrieNode(char val) {
-            this.val = val;
-            this.children = new TrieNode[26];
-            this.isWord = false;
+            public TrieNode(char val) {
+                this.val = val;
+                this.children = new TrieNode[26];
+                this.isWord = false;
+            }
         }
     }
 }

src/test/java/com/fishercoder/_648Test.java

@@ -10,20 +10,20 @@
 import static org.junit.Assert.assertEquals;
 
 public class _648Test {
-    private static _648 test;
+    private static _648.Solution1 solution1;
     private static List<String> dict;
     private static String sentence;
 
     @BeforeClass
     public static void setup() {
-        test = new _648();
+        solution1 = new _648.Solution1();
     }
 
     @Test
     public void test1() {
         dict = Arrays.asList("cat", "bat", "rat");
         sentence = "the cattle was rattled by the battery";
-        assertEquals("the cat was rat by the bat", test.replaceWords(dict, sentence));
+        assertEquals("the cat was rat by the bat", solution1.replaceWords(dict, sentence));
     }
 
     @Test
@@ -57,7 +57,7 @@ public void test2() {
                 + "dijhrrpnwjlju muzzrrsypzgwvblf z h q i daee r nlipyfszvxlwqw "
                 + "yoq dewpgtcrzausqwhh q i k bqprarpgnyemzwifqzz "
                 + "oai pnqottd nygesjtlpala q gyvukjpc s mxhlkdaycskj "
-                + "uvwmerplaibeknltuvd ocnn f c pxbd oklwhcppuziixpvihihp", test.replaceWords(dict, sentence));
+                + "uvwmerplaibeknltuvd ocnn f c pxbd oklwhcppuziixpvihihp", solution1.replaceWords(dict, sentence));
     }
 
 }