[LeetCode] 1165. Single-Row Keyboard

1165. Single-Row Keyboard

Easy

There is a special keyboard with all keys in a single row.

Given a string keyboard of length 26 indicating the layout of the keyboard (indexed from 0 to 25). Initially, your finger is at index 0. To type a character, you have to move your finger to the index of the desired character. The time taken to move your finger from index i to index j is |i - j|.

You want to type a string word. Write a function to calculate how much time it takes to type it with one finger.

 

Example 1:

Input: keyboard = "abcdefghijklmnopqrstuvwxyz", word = "cba"
Output: 4
Explanation: The index moves from 0 to 2 to write 'c' then to 1 to write 'b' then to 0 again to write 'a'.
Total time = 2 + 1 + 1 = 4. 

Example 2:

Input: keyboard = "pqrstuvwxyzabcdefghijklmno", word = "leetcode"
Output: 73

 

Constraints:

• keyboard.length == 26
• keyboard contains each English lowercase letter exactly once in some order.
• 1 <= word.length <= 104
• word[i] is an English lowercase letter.

---

문제 풀이

• 키보드 배열과 단어하나가 주어진다.
• 주어진 단어를 키보드 하나하나 입력할때 주어진 키보드 배열에서 인덱스를 옮기면서 얼마나 걸리는지 구해야한다.
• 키보드 배열은 임의로 주어지기 때문에 딕셔너리에 인덱스와 함께 저장한다.
• 그리고 첫 글자는 인덱스를 바로 더하고 현재 인덱스로 저장한다. 
• 그리고 다음 인덱스부터 현재 위치에서 다음 알파벳의 위치의 거리 차이를 계산하여 정답에 더한다.
• 이때도 마찬가지로 현재 위치를 갱신한다.

소스 코드

class Solution:
    def calculateTime(self, keyboard: str, word: str) -> int:
        ans = 0
        d = dict()
        cur = 0
        for i, k in enumerate(keyboard):
            d[k] = i
        ans += d[word[0]]
        cur = d[word[0]]
        for i in range(1, len(word)):
            ans += abs(cur - d[word[i]])
            cur = d[word[i]]
        return ans