Update Goat Latin.java

varunu28 · web-flow · commit bd9598f7ed8a · 2022-09-09T10:25:09.000-07:00
diff --git a/Easy/Goat Latin.java b/Easy/Goat Latin.java
@@ -1,29 +1,19 @@
 class Solution {
-  public String toGoatLatin(String S) {
-    StringBuilder sb = new StringBuilder();
-    StringBuilder endAppend = new StringBuilder("a");
-    final String CONSTANT_APPEND = "ma";
-    int start = 0;
-    int end = 0;
-    int n = S.length();
-    while (end < n) {
-      while (end < n && S.charAt(end) != ' ') {
-        end++;
-      }
-      char c = Character.toLowerCase(S.charAt(start));
-      if (c == 'a' || c == 'e' || c == 'i' || c == 'o' || c == 'u') {
-        sb.append(S.substring(start, end)).append(CONSTANT_APPEND).append(endAppend.toString());
-      }
-      else {
-        sb.append(S.substring(start + 1, end)).append(S.charAt(start)).append(CONSTANT_APPEND).append(endAppend.toString());
-      }
-      endAppend.append('a');
-      end++;
-      start = end;
-      if (end < n) {
-        sb.append(' ');
+  public String toGoatLatin(String sentence) {
+    String[] words = sentence.split("\\s+");
+    StringBuilder result = new StringBuilder();
+    StringBuilder aWord = new StringBuilder();
+    Set<Character> vowels = Set.of('a', 'e', 'i', 'o', 'u');
+    for (String word : words) {
+      aWord.append('a');
+      char firstChar = word.charAt(0);
+      if (vowels.contains(Character.toLowerCase(firstChar))) {
+        result.append(word).append("ma");
+      } else {
+        result.append(word.substring(1)).append(firstChar).append("ma");
       }
+      result.append(aWord.toString()).append(" ");
     }
-    return sb.toString();
+    return result.toString().trim();
   }
 }
