leetcode541

Posted 2021-02-07 yxlsblog

题目：

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

示例:

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

要求:

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

解决方案：

直接对给定字符串s中的1，3，5，7...个长度为k的小字符串进行反转即可。

解决代码：

　　

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

相关知识和程序bug：

1.reverse（first,last）函数：

函数功能为反转范围里的元素（reverse range),两个参数均为迭代器。

2.如果len<k：

此处代码为i<=count，这样可能会超出s的范围。但如果代码是i<count，则当len<k时，不会执行循环，从而会返回原字符串，导致错误。