add a solution for 1209

fishercoder1534 · fishercoder1534 · commit 74c47d112c43 · 2021-04-16T08:04:08.000-07:00
diff --git a/src/main/java/com/fishercoder/solutions/_1209.java b/src/main/java/com/fishercoder/solutions/_1209.java
@@ -43,4 +43,33 @@ public String removeDuplicates(String s, int k) {
             return sb.reverse().toString();
         }
     }
+
+    public static class Solution2 {
+        public String removeDuplicates(String s, int k) {
+            StringBuilder sb = new StringBuilder();
+            int dupCount = 0;
+            for (int i = 0; i < s.length(); i++) {
+                if (sb.length() != 0 && sb.charAt(sb.length() - 1) == s.charAt(i)) {
+                    dupCount++;
+                } else {
+                    dupCount = 1;
+                }
+                sb.append(s.charAt(i));
+                if (dupCount == k) {
+                    sb.setLength(sb.length() - k);
+                    if (i + 1 < s.length()) {
+                        dupCount = 0;
+                        for (int j = sb.length() - 1; j >= 0; j--) {
+                            if (sb.charAt(j) == s.charAt(i + 1)) {
+                                dupCount++;
+                            } else {
+                                break;
+                            }
+                        }
+                    }
+                }
+            }
+            return sb.toString();
+        }
+    }
 }
diff --git a/src/test/java/com/fishercoder/_1209Test.java b/src/test/java/com/fishercoder/_1209Test.java
@@ -9,25 +9,36 @@
 public class _1209Test {
 
     private static _1209.Solution1 solution1;
+    private static _1209.Solution2 solution2;
 
     @BeforeClass
     public static void setup() {
         solution1 = new _1209.Solution1();
+        solution2 = new _1209.Solution2();
     }
 
     @Test
     public void test1() {
         assertEquals("abcd", solution1.removeDuplicates("abcd", 2));
+        assertEquals("abcd", solution2.removeDuplicates("abcd", 2));
     }
 
     @Test
     public void test2() {
         assertEquals("aa", solution1.removeDuplicates("deeedbbcccbdaa", 3));
+        assertEquals("aa", solution2.removeDuplicates("deeedbbcccbdaa", 3));
     }
 
     @Test
     public void test3() {
         assertEquals("ps", solution1.removeDuplicates("pbbcggttciiippooaais", 2));
+        assertEquals("ps", solution2.removeDuplicates("pbbcggttciiippooaais", 2));
+    }
+
+    @Test
+    public void test4() {
+        assertEquals("ghayqgq", solution1.removeDuplicates("ghanyhhhhhttttttthhyyyyyynnnnnnyqkkkkkkkrrrrrrjjjjjjjryyyyyyfffffffygq", 7));
+        assertEquals("ghayqgq", solution2.removeDuplicates("ghanyhhhhhttttttthhyyyyyynnnnnnyqkkkkkkkrrrrrrjjjjjjjryyyyyyfffffffygq", 7));
     }
 
 }
