Update Single-Row Keyboard.java

varunu28 · web-flow · commit 290f75fa9e8d · 2024-01-15T13:13:06.000-08:00
diff --git a/Easy/Single-Row Keyboard.java b/Easy/Single-Row Keyboard.java
@@ -1,16 +1,15 @@
 class Solution {
-  public int calculateTime(String keyboard, String word) {
-    Map<Character, Integer> map = new HashMap<>();
-    int pos = 0;
-    for (char c : keyboard.toCharArray()) {
-      map.put(c, pos++);
+    public int calculateTime(String keyboard, String word) {
+        Map<Character, Integer> map = new HashMap<>();
+        for (int i = 0; i < 26; i++) {
+            map.put(keyboard.charAt(i), i);
+        }
+        int position = 0;
+        int time = 0;
+        for (char c : word.toCharArray()) {
+            time += Math.abs(position - map.get(c));
+            position = map.get(c);
+        }
+        return time;
     }
-    int total = 0;
-    int currPos = 0;
-    for (char c : word.toCharArray()) {
-      total += Math.abs(currPos - map.get(c));
-      currPos = map.get(c);
-    }
-    return total;
-  }
 }
