力扣--541反转字符串 II

题目描述
给定一个字符串和一个整数 k，你需要对从字符串开头算起的每个 2k 个字符的前k个字符进行反转。如果剩余少于 k 个字符，则将剩余的所有全部反转。如果有小于 2k 但大于或等于 k 个字符，则反转前 k 个字符，并将剩余的字符保持原样。

示例:

输入: s = “abcdefg”, k = 2
输出: “bacdfeg”

要求:

该字符串只包含小写的英文字母。
给定字符串的长度和 k 在[1, 10000]范围内。

思路
将题目的三句话拆成三个if条件即可。定义一个标志p，初始值为0，通过它找到当前访问的元素位置。

1. 如果剩余字符大于2k个，则利用reverse函数执行翻转。每次起点为p+begin(),终点为p+k+begin()。
2. 如果剩余少于 k 个字符，则翻转全部，起点为p+begin(),终点为end().
3. 如果有小于 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;
    }
};