Update One Edit Distance.java

varunu28 · web-flow · commit c81691fc1689 · 2022-02-14T11:23:27.000-08:00
diff --git a/Medium/One Edit Distance.java b/Medium/One Edit Distance.java
@@ -1,50 +1,39 @@
 class Solution {
-    public static boolean isOneEditDistance(String s, String t) {
-        if (Math.abs(s.length() - t.length()) > 1) {
-            return false;
-        }
-        else if (s.length() - t.length() == 0) {
-            if (s.equals(t)) {
-                return false;
-            }
-
-            boolean missed = false;
-            for (int i=0; i<s.length(); i++) {
-                if (s.charAt(i) == t.charAt(i)) {
-                    continue;
-                }
-                if (missed) {
-                    return false;
-                }
-                
-                missed = true;
-            }
-            return true;
-        }
-
-        return s.length() - t.length() == 1 ? oneDeleteOnly(s,t) : oneDeleteOnly(t,s);
+  public boolean isOneEditDistance(String s, String t) {
+    int lengthDiff = Math.abs(s.length() - t.length());
+    if (s.equals(t) || lengthDiff > 1) {
+      return false;
     }
-
-    private static boolean oneDeleteOnly(String s, String t) {
-        if (t.length() == 0) {
-            return true;
-        }
-        int i = 0;
-        int j = 0;
-
-        while (i < s.length() && j < t.length()) {
-            if (s.charAt(i) != t.charAt(j)) {
-                if (i > j) {
-                    return false;
-                }
-                i++;
-                continue;
-            }
-
-            i++;
-            j++;
+    if (s.length() == 0 || t.length() == 0) {
+      return true;
+    }
+    int idxOne = 0;
+    int idxTwo = 0;
+    boolean changeDone = false;
+    while (idxOne < s.length() && idxTwo < t.length()) {
+      if (s.charAt(idxOne) == t.charAt(idxTwo)) {
+        idxOne++;
+        idxTwo++;
+        continue;
+      }
+      if (changeDone) {
+        return false;
+      }
+      if (lengthDiff != 0) {
+        if (s.length() > t.length()) {
+          idxOne++;
+        } else {
+          idxTwo++;
         }
-
-        return true;
+      } else {
+        idxOne++;
+        idxTwo++;
+      }
+      changeDone = true;
+    }
+    if (changeDone && (idxOne != s.length() || idxTwo != t.length())) {
+      return false;
     }
+    return true;
+  }
 }
