Question

1427. 字符串的左右移

English Version

题目描述

给定一个包含小写英文字母的字符串 s 以及一个矩阵 shift，其中 shift[i] = [direction, amount]：

• direction 可以为 0 （表示左移）或 1 （表示右移）。
• amount 表示 s 左右移的位数。
• 左移 1 位表示移除 s 的第一个字符，并将该字符插入到 s 的结尾。
• 类似地，右移 1 位表示移除 s 的最后一个字符，并将该字符插入到 s 的开头。

对这个字符串进行所有操作后，返回最终结果。

 

示例 1：

输入：s = "abc", shift = [[0,1],[1,2]]
输出："cab"
解释：
[0,1] 表示左移 1 位。 "abc" -> "bca"
[1,2] 表示右移 2 位。 "bca" -> "cab"

示例 2：

输入：s = "abcdefg", shift = [[1,1],[1,1],[0,2],[1,3]]
输出："efgabcd"
解释： 
[1,1] 表示右移 1 位。 "abcdefg" -> "gabcdef"
[1,1] 表示右移 1 位。 "gabcdef" -> "fgabcde"
[0,2] 表示左移 2 位。 "fgabcde" -> "abcdefg"
[1,3] 表示右移 3 位。 "abcdefg" -> "efgabcd"

 

提示：

• 1 <= s.length <= 100
• s 只包含小写英文字母
• 1 <= shift.length <= 100
• shift[i].length == 2
• 0 <= shift[i][0] <= 1
• 0 <= shift[i][1] <= 100

解法

方法一：模拟

我们不妨记字符串 $s$ 的长度为 $n$。接下来遍历数组 $shift$，累加得到最终的偏移量 $x$，然后将 $x$ 对 $n$ 取模，最终结果就是将 $s$ 的前 $n - x$ 个字符移动到末尾。

时间复杂度 $O(n + m)$，其中 $n$ 和 $m$ 分别是字符串 $s$ 的长度和数组 $shift$ 的长度。空间复杂度 $O(1)$。

class Solution:
  def stringShift(self, s: str, shift: List[List[int]]) -> str:
    x = sum((b if a else -b) for a, b in shift)
    x %= len(s)
    return s[-x:] + s[:-x]

class Solution {
  public String stringShift(String s, int[][] shift) {
    int x = 0;
    for (var e : shift) {
      if (e[0] == 0) {
        e[1] *= -1;
      }
      x += e[1];
    }
    int n = s.length();
    x = (x % n + n) % n;
    return s.substring(n - x) + s.substring(0, n - x);
  }
}

class Solution {
public:
  string stringShift(string s, vector<vector<int>>& shift) {
    int x = 0;
    for (auto& e : shift) {
      if (e[0] == 0) {
        e[1] = -e[1];
      }
      x += e[1];
    }
    int n = s.size();
    x = (x % n + n) % n;
    return s.substr(n - x, x) + s.substr(0, n - x);
  }
};

func stringShift(s string, shift [][]int) string {
  x := 0
  for _, e := range shift {
    if e[0] == 0 {
      e[1] = -e[1]
    }
    x += e[1]
  }
  n := len(s)
  x = (x%n + n) % n
  return s[n-x:] + s[:n-x]
}

function stringShift(s: string, shift: number[][]): string {
  let x = 0;
  for (const [a, b] of shift) {
    x += a === 0 ? -b : b;
  }
  x %= s.length;
  if (x < 0) {
    x += s.length;
  }
  return s.slice(-x) + s.slice(0, -x);
}