给定一个包含小写英文字母的字符串 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
方法一:模拟
遍历 shift
,累加(减)得到最终偏移量 s
进行左移或右移。
时间复杂度 s
的长度,$m$ 为 shift
的长度。
class Solution:
def stringShift(self, s: str, shift: List[List[int]]) -> str:
x = 0
for a, b in shift:
if a == 0:
b = -b
x += b
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] = -e[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]
}