leetcode541. 反转字符串II

题目描述

给定一个字符串 s 和一个整数 k，从字符串开头算起，每计数至 2k 个字符，就反转这 2k 字符中的前 k 个字符。

如果剩余字符少于 k 个，则将剩余字符全部反转。
如果剩余字符小于 2k 但大于或等于 k 个，则反转前 k 个字符，其余字符保持原样。

示例 1：

输入：s = "abcdefg", k = 2
输出："bacdfeg"

示例 2：

输入：s = "abcd", k = 2
输出："bacd"

提示：

1 <= s.length <= 104
s 仅由小写英文组成
1 <= k <= 104

分析思路

在for循环中要每隔2k的跳一次。进行判断的时候，首先判断i+k是否小于等于len长度，如果不是的话， 就操作了空的字符。
这里用到了c++的库函数reverse，原理是传入两个下标，把区间内的字符串进行反转。具体代码如下：

class Solution {
public:
    string reverseStr(string s, int k) {
        int len = s.size();
        for(int i=0;i<len;i+=(2*k)){
            if(i+k <= len){
                // 这时候的反转，对应了题干中的规则和下面的剩余字符小于2k但大于等于k个，则反转前k个字符
                reverse(s.begin()+i, s.begin()+i+k);
                // 然后直接进行continue，进入下一次循环
                continue;
            }
            // 处理剩余的字符串，如果小于k个，则全部反转
            reverse(s.begin()+i, s.end());
            
        }
        return s;
    }
};