541. 反转字符串 II

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

示例:

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

要求:

  1. 该字符串只包含小写的英文字母。
  2. 给定字符串的长度和 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;
    }
};

 

你可能感兴趣的:(LeetCode)