题目描述
给定一个字符串和一个整数 k,你需要对从字符串开头算起的每个 2k 个字符的前k个字符进行反转。如果剩余少于 k 个字符,则将剩余的所有全部反转。如果有小于 2k 但大于或等于 k 个字符,则反转前 k 个字符,并将剩余的字符保持原样。
样例
输入: s = "abcdefg", k = 2
输出: "bacdfeg"
要求
- 该字符串只包含小写的英文字母。
- 给定字符串的长度和 k 在[1, 10000]范围内。
算法
(模拟) $O(n)$
- 第一层循环枚举每 2k 个字母的开头位置。
- 第二层循环根据第一层循环做交换操作,注意判断交换的末尾是否超过了字符串的长度。
时间复杂度
- 每个字母最多仅被遍历一次,故时间复杂度为 $O(n)$。
C++ 代码
class Solution {
public:
string reverseStr(string s, int k) {
int n = s.length();
for (int i = 0; i < n; i += 2 * k)
for (int st = i, ed = min(n, i + k) - 1; st < ed; st++, ed--)
swap(s[st], s[ed]);
return s;
}
};