LeetCode-Python-1165. 单行键盘

我们定制了一款特殊的力扣键盘，所有的键都排列在一行上。

我们可以按从左到右的顺序，用一个长度为 26 的字符串 keyboard （索引从 0 开始，到 25 结束）来表示该键盘的键位布局。

现在需要测试这个键盘是否能够有效工作，那么我们就需要个机械手来测试这个键盘。

最初的时候，机械手位于左边起第一个键（也就是索引为 0 的键）的上方。当机械手移动到某一字符所在的键位时，就会在终端上输出该字符。

机械手从索引 i 移动到索引 j 所需要的时间是 |i - j|。

当前测试需要你使用机械手输出指定的单词 word，请你编写一个函数来计算机械手输出该单词所需的时间。

提示：

• keyboard.length == 26
• keyboard 按某种特定顺序排列，并包含每个小写英文字母一次。
• 1 <= word.length <= 10^4
• word[i] 是一个小写英文字母

思路：

热身题，需要多次查找字母的位置，所以用哈希表。

先开个哈希表记录一下每个字母的下标，用一个变量last_pos记录机械手当前在的位置，

然后线性扫描需要输出的单词，通过读哈希表找到机械手需要移动到的位置，

把移动的距离加到答案里，然后刷新last_pos即可。

class Solution(object):
    def calculateTime(self, keyboard, word):
        """
        :type keyboard: str
        :type word: str
        :rtype: int
        """
        dic = dict()
        for i, char in enumerate(keyboard):
            dic[char] = i
            
        res, last_pos = 0, 0
        for char in word:
            res += abs(dic[char] - last_pos)
            last_pos = dic[char]           
        return res