- 单行键盘
题目:
我们定制了一款特殊的力扣键盘,所有的键都排列在一行上。
我们可以按从左到右的顺序,用一个长度为 26 的字符串 keyboard (索引从 0 开始,到 25 结束)来表示该键盘的键位布局。
现在需要测试这个键盘是否能够有效工作,那么我们就需要个机械手来测试这个键盘。
最初的时候,机械手位于左边起第一个键(也就是索引为 0 的键)的上方。当机械手移动到某一字符所在的键位时,就会在终端上输出该字符。
机械手从索引 i 移动到索引 j 所需要的时间是 |i - j|。
当前测试需要你使用机械手输出指定的单词 word,请你编写一个函数来计算机械手输出该单词所需的时间。
示例 1:
输入:keyboard = “abcdefghijklmnopqrstuvwxyz”, word = “cba”
输出:4
解释:
机械手从 0 号键移动到 2 号键来输出 ‘c’,又移动到 1 号键来输出 ‘b’,接着移动到 0 号键来输出 ‘a’。
总用时 = 2 + 1 + 1 = 4.
示例 2:
输入:keyboard = “pqrstuvwxyzabcdefghijklmno”, word = “leetcode”
输出:73
提示:
keyboard.length == 26
keyboard 按某种特定顺序排列,并包含每个小写英文字母一次。
1 <= word.length <= 10^4
word[i] 是一个小写英文字母
解答:
(1)
首先建立一个字典,将键盘的字母和键盘的位置组成键值对存储到字典中,然后遍历词,让词中的每一个字母在字典中查找,利用后一个字母和前一个字母位置的差值的绝对值计算步数,最后再加上初始第一个字母需要移动的步数即可。
class Solution:
def calculateTime(self, keyboard: str, word: str) -> int:
reuslt = 0
keyboard_dict = {}
for i in range(len(keyboard)):
keyboard_dict[keyboard[i]] = i
for j in range(len(word)-1):
reuslt += abs((keyboard_dict[word[j]] - keyboard_dict[word[j+1]]))
reuslt += keyboard_dict[word[0]]
return reuslt
(2)使用zip函数打包字典,keyboard中的字母作为index,位置作为value,其余部分与方法1相同
class Solution:
def calculateTime(self, keyboard: str, word: str) -> int:
reuslt = 0
keyboard_dict = dict(zip(keyboard,[i for i in range (len(keyboard))]))
for j in range(len(word)-1):
reuslt += abs((keyboard_dict[word[j]] - keyboard_dict[word[j+1]]))
reuslt += keyboard_dict[word[0]]
return reuslt