题目描述
给定一个字符串和一个整数 k,你需要对从字符串开头算起的每个 2k 个字符的前k个字符进行反转。如果剩余少于 k 个字符,则将剩余的所有全部反转。如果有小于 2k 但大于或等于 k 个字符,则反转前 k 个字符,并将剩余的字符保持原样。
示例:
输入: s = “abcdefg”, k = 2
输出: “bacdfeg”
要求:
该字符串只包含小写的英文字母。
给定字符串的长度和 k 在[1, 10000]范围内。
思路
将题目的三句话拆成三个if条件即可。定义一个标志p,初始值为0,通过它找到当前访问的元素位置。
- 如果剩余字符大于2k个,则利用reverse函数执行翻转。每次起点为p+begin(),终点为p+k+begin()。
- 如果剩余少于 k 个字符,则翻转全部,起点为p+begin(),终点为end().
- 如果有小于 2k 但大于或等于 k 个字符,与(1)一样。
踩过的坑
第一次写的时候手残在end()后面-1,注意reverse()函数的参数!它的参数就是begin()和end()!可以将begin()和end()区间的元素全部逆置,所以不要像我一样画蛇添足。
源代码
class Solution {
public:
string reverseStr(string s, int k) {
if(s.empty())
{
return nullptr;
}
int size=s.size();
int p=0;
while(p<size)
{
if(p+2*k<=size)
{
reverse(s.begin()+p,s.begin()+p+k);
}
if(p+k>size)
{
reverse(s.begin()+p,s.end());
}
if(p+2*k>size&&p+k<=size)
{
reverse(s.begin()+p,s.begin()+p+k);
}
p+=2*k;
}
return s;
}
};