【简单】541.反转字符串Ⅱ

题目描述

给定一个字符串 s 和一个整数 k，从字符串开头算起，每计数至 2k 个字符，就反转这 2k 字符中的前 k 个字符。

如果剩余字符少于 k 个，则将剩余字符全部反转。
如果剩余字符小于 2k 但大于或等于 k 个，则反转前 k 个字符，其余字符保持原样。

示例 1：

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

示例 2：

输入：s = "abcd", k = 2
输出："bacd"

解题思路

来源于：代码随想录

其实在遍历字符串的过程中，只要让 i += (2 * k)，i 每次移动 2 * k 就可以了，然后判断是否需要有反转的区间。

因为要找的也就是每2 * k 区间的起点，这样写，程序会高效很多。

那么这里具体反转的逻辑我们要不要使用库函数呢，其实用不用都可以，使用reverse来实现反转也没毛病，毕竟不是解题关键部分。

代码如下：

class Solution {
public:
    string reverseStr(string s, int k) {
        for(int i = 0; i < s.size(); i += (2 * k))
        {
            // 1. 每隔 2k 个字符的前 k 个字符进行反转
            // 2. 剩余字符小于 2k 但大于或等于 k 个，则反转前 k 个字符
            if(i + k <= s.size())
            {
                reverse(s.begin() + i, s.begin() + i + k);
            }

            // 3. 剩余字符少于 k 个，则将剩余字符全部反转。
            else
            {
                reverse(s.begin() + i, s.end());
            }
        }
        return s;
    }
};

时间复杂度: O(n)
空间复杂度: O(1)

其实还有其他的解法，但是我觉得掌握一种简单的，自己能理解的一种方法就可以了，如果你们想掌握更多的方法，可以去看代码随想录