add one more solution for 821

fishercoder1534 · fishercoder1534 · commit 69e3c2751f04 · 2021-02-08T08:26:09.000-08:00
diff --git a/src/main/java/com/fishercoder/solutions/_821.java b/src/main/java/com/fishercoder/solutions/_821.java
@@ -1,5 +1,6 @@
 package com.fishercoder.solutions;
 
+import java.util.Arrays;
 import java.util.TreeSet;
 
 public class _821 {
@@ -27,4 +28,35 @@ public int[] shortestToChar(String S, char C) {
             return result;
         }
     }
+
+    public static class Solution2 {
+        public int[] shortestToChar(String s, char c) {
+            int[] result = new int[s.length()];
+            Arrays.fill(result, Integer.MAX_VALUE);
+            for (int i = 0; i < s.length(); i++) {
+                if (s.charAt(i) == c) {
+                    result[i] = 0;
+                }
+            }
+            for (int i = 0; i < s.length(); i++) {
+                if (result[i] != 0) {
+                    int j = i - 1;
+                    while (j >= 0 && result[j] != 0) {
+                        j--;
+                    }
+                    if (j >= 0) {
+                        result[i] = i - j;
+                    }
+                    j = i + 1;
+                    while (j < s.length() && result[j] != 0) {
+                        j++;
+                    }
+                    if (j < s.length()) {
+                        result[i] = Math.min(result[i], j - i);
+                    }
+                }
+            }
+            return result;
+        }
+    }
 }
diff --git a/src/test/java/com/fishercoder/_821Test.java b/src/test/java/com/fishercoder/_821Test.java
@@ -7,17 +7,25 @@
 import static org.junit.Assert.assertArrayEquals;
 
 public class _821Test {
-  private static _821.Solution1 solution1;
-  private static int[] expected;
+    private static _821.Solution1 solution1;
+    private static _821.Solution2 solution2;
+    private static int[] expected;
 
-  @BeforeClass
-  public static void setup() {
-    solution1 = new _821.Solution1();
-  }
+    @BeforeClass
+    public static void setup() {
+        solution1 = new _821.Solution1();
+        solution2 = new _821.Solution2();
+    }
 
-  @Test
-  public void test1() {
-    expected = new int[] {3, 2, 1, 0, 1, 0, 0, 1, 2, 2, 1, 0};
-    assertArrayEquals(expected, solution1.shortestToChar("loveleetcode", 'e'));
-  }
+    @Test
+    public void test1() {
+        expected = new int[]{3, 2, 1, 0, 1, 0, 0, 1, 2, 2, 1, 0};
+        assertArrayEquals(expected, solution1.shortestToChar("loveleetcode", 'e'));
+    }
+
+    @Test
+    public void test2() {
+        expected = new int[]{3, 2, 1, 0, 1, 0, 0, 1, 2, 2, 1, 0};
+        assertArrayEquals(expected, solution2.shortestToChar("loveleetcode", 'e'));
+    }
 }
