Question

848. 字母移位

English Version

题目描述

有一个由小写字母组成的字符串 s，和一个长度相同的整数数组 shifts。

我们将字母表中的下一个字母称为原字母的 _移位_ shift() （由于字母表是环绕的， 'z' 将会变成 'a'）。

• 例如，shift('a') = 'b', shift('t') = 'u', 以及 shift('z') = 'a'。

对于每个 shifts[i] = x ， 我们会将 s 中的前 i + 1 个字母移位 x 次。

返回 _将所有这些移位都应用到 s 后最终得到的字符串_ 。

 

示例 1：

输入：s = "abc", shifts = [3,5,9]
输出："rpl"
解释： 
我们以 "abc" 开始。
将 S 中的第 1 个字母移位 3 次后，我们得到 "dbc"。
再将 S 中的前 2 个字母移位 5 次后，我们得到 "igc"。
最后将 S 中的这 3 个字母移位 9 次后，我们得到答案 "rpl"。

示例 2:

输入: s = "aaa", shifts = [1,2,3]
输出: "gfd"

 

提示:

• 1 <= s.length <= 105
• s 由小写英文字母组成
• shifts.length == s.length
• 0 <= shifts[i] <= 109

​​​​​​

解法

方法一：后缀和

对于字符串 $s$ 中的每个字符，我们需要计算其最终的偏移量，即 shifts[i] 与 shifts[i + 1] 与 shifts[i + 2] … 的和。我们可以使用后缀和的思想，从后往前遍历 shifts，计算每个字符的最终偏移量，然后对 $26$ 取模，得到最终的字符。

时间复杂度 $O(n)$，其中 $n$ 为字符串 $s$ 的长度。忽略答案的空间消耗，空间复杂度 $O(1)$。

class Solution:
  def shiftingLetters(self, s: str, shifts: List[int]) -> str:
    n, t = len(s), 0
    s = list(s)
    for i in range(n - 1, -1, -1):
      t += shifts[i]
      j = (ord(s[i]) - ord('a') + t) % 26
      s[i] = ascii_lowercase[j]
    return ''.join(s)

class Solution:
  def shiftingLetters(self, s: str, shifts: List[int]) -> str:
    n = len(s)
    d = [0] * (n + 1)
    for i, c in enumerate(s):
      v = ord(c) - ord('a')
      d[i] += v
      d[i + 1] -= v
    for i, x in enumerate(shifts):
      d[0] += x
      d[i + 1] -= x
    t = 0
    ans = []
    for i in range(n):
      d[i] %= 26
      ans.append(ascii_lowercase[d[i]])
      d[i + 1] += d[i]
    return ''.join(ans)

class Solution {
  public String shiftingLetters(String s, int[] shifts) {
    char[] cs = s.toCharArray();
    int n = cs.length;
    long t = 0;
    for (int i = n - 1; i >= 0; --i) {
      t += shifts[i];
      int j = (int) ((cs[i] - 'a' + t) % 26);
      cs[i] = (char) ('a' + j);
    }
    return String.valueOf(cs);
  }
}

class Solution {
public:
  string shiftingLetters(string s, vector<int>& shifts) {
    long long t = 0;
    int n = s.size();
    for (int i = n - 1; ~i; --i) {
      t += shifts[i];
      int j = (s[i] - 'a' + t) % 26;
      s[i] = 'a' + j;
    }
    return s;
  }
};

func shiftingLetters(s string, shifts []int) string {
  t := 0
  n := len(s)
  cs := []byte(s)
  for i := n - 1; i >= 0; i-- {
    t += shifts[i]
    j := (int(cs[i]-'a') + t) % 26
    cs[i] = byte('a' + j)
  }
  return string(cs)
}