题目地址:
https://leetcode.com/problems/shifting-letters/
给定一个小写英文字母的长
n
n
n字符串
s
s
s,和一个等长的非负整数数组
A
A
A,要求先将
s
[
0
]
s[0]
s[0]字符增加
A
[
0
]
A[0]
A[0]这么多数,然后将
s
[
0
:
1
]
s[0:1]
s[0:1]增加
A
[
1
]
A[1]
A[1]这么多数,以此类推,如果出了'z'
,则继续从'a'
开始数起。
先求 A A A的后缀和,这样就求出了 s [ i ] s[i] s[i]要增加多少,接着做变换即可。这里需要注意一些边界条件,防止出界,要多次模 26 26 26。代码如下:
public class Solution {
public String shiftingLetters(String s, int[] shifts) {
shifts[shifts.length - 1] %= 26;
for (int i = shifts.length - 2; i >= 0; i--) {
shifts[i] += shifts[i + 1] % 26;
shifts[i] %= 26;
}
char[] chs = s.toCharArray();
for (int i = 0; i < chs.length; i++) {
chs[i] += shifts[i];
while (chs[i] > 'z') {
chs[i] -= 26;
}
}
return new String(chs);
}
}
时空复杂度 O ( n ) O(n) O(n)。