【LeetCode】541. 反转字符串 II

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/fuqiuai/article/details/83117475

题目链接：https://leetcode-cn.com/problems/reverse-string-ii/description/

题目描述

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

示例

输入: s = “abcdefg”, k = 2
输出: “bacdfeg”

要求:

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

解决方法

题目较简单

class Solution {
public:
    string reverseStr(string s, int k) {
        if (s.size()<=1 || k<=1) return s; //特殊情况
        int left=0;
        int len=s.size();
        while(left<=len-2*k){
            reverse(s.begin()+left,s.begin()+left+k);
            left+=2*k;
        }
        int res=s.size()-left;//剩下的字符数
        if (res<=k) reverse(s.begin()+left,s.end()); //情况1：剩余少于 k 个字符
        else reverse(s.begin()+left,s.begin()+left+k); //情况2：剩余小于 2k 但大于或等于 k 个字符
        
        return s;
    }
};