848. 字母移位
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/shifting-letters
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
题目描述
有一个由小写字母组成的字符串 S
,和一个整数数组 shifts
。
我们将字母表中的下一个字母称为原字母的 移位(由于字母表是环绕的, 'z'
将会变成 'a'
)。
例如·,shift('a') = 'b'
, shift('t') = 'u',
, 以及 shift('z') = 'a'
。
对于每个 shifts[i] = x
, 我们会将 S
中的前 i+1
个字母移位 x
次。
返回将所有这些移位都应用到 S
后最终得到的字符串。
示例
输入:S = “abc”, shifts = [3,5,9]
输出:“rpl”
解释:
我们以 “abc” 开始。
将 S 中的第 1 个字母移位 3 次后,我们得到 “dbc”。
再将 S 中的前 2 个字母移位 5 次后,我们得到 “igc”。
最后将 S 中的这 3 个字母移位 9 次后,我们得到答案 “rpl”。
提示
- 1 <= S.length = shifts.length <= 20000
- 0 <= shifts[i] <= 10 ^ 9
解题思路
对于输入序列S中的字母,第i次移动将会移动他前面的所有字符,由于shifts长度与S相同,因此,最简单的情况是最后一个字符,他只移动一次,距离为shifts[N],倒数第二个字符移动了两次,距离为shifts[N]+shifts[N+1],第一个字符移动了sum(shifts)的距离。
故题目可以分解成两步,即
- 找到当前字母的index
- 对字母移动相应距离
容易想到,从最后一个字符开始移动,然后依次累加移动距离即可,但由于最终结果是一个拼接而成的字符串,我习惯从第一个开始往后拼的写法,所以是先求了移动总长,然后依次做减法的写法。
代码
第一次写是使用了一个list存放26个字母,来实现第一步“找到当前字母index”的方案,代码如下:
class Solution:
def shiftingLetters(self, S: str, shifts: List[int]) -> str:
letters = ["a","b","c","d","e","f","g","h","i","j","k","l","m","n","o","p","q","r","s","t","u","v","w","x","y","z"]
n = 0
for i in shifts:
n+=i
res = ""
for j, letter in enumerate(S):
letter_index = letters.index(letter)
if j != 0:
n = n-shifts[j-1]
res+=letters[(letter_index+n)%26]
return res
后来觉得第一次的方法太麻烦了,直接使用ascii码来记录index,这里学习了几个ascii码相关的知识:
ord()
返回字符对应的ascii码,如ord("a") = 97
chr()
返回ascii码对应的字符,如chr(97) = "a"
class Solution:
def shiftingLetters(self, S: str, shifts: List[int]) -> str:
n = sum(shifts)
res = ""
for j, letter in enumerate(S):
ascii = ord(letter)
if j != 0:
n = n-shifts[j-1]
res+=chr((ascii+n-97)%26+97)
return res
不得不说修改之后看起来确实简洁了不少