541. 反转字符串 II-蒲公英云

541. 反转字符串 II

悠悠 2022-02-02 12:15 303阅读 0赞

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

示例:

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

要求:

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

【分析】

这道题的意思其实是每隔k隔字符，翻转k个字符，最后如果不够k个了的话，则全部翻转。比较直接的方法就是先用n／k算出来原字符串s能分成几个长度为k的字符串，然后开始遍历这些字符串，遇到2的倍数就翻转，翻转的时候注意考虑下是否已经到s末尾了，参见代码如下：

class Solution {
public:
    string reverseStr(string s, int k) {
        int n = s.size();
        int count = n/k;
        for(int i = 0; i <= count; ++i){
            if( i % 2 == 0){
                if(i * k + k < n)
                    reverse(s.begin() + i * k,s.begin() + i * k + k);
                else
                    reverse(s.begin() + i * k,s.end());//到达末尾
            }
        }
        return s;
    }
};