力扣--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;
    }
};

你可能感兴趣的:(刷题,C++,反转字符串)