LeetCode_541 反转字符串II

216 阅读1分钟

1、题目:反转字符串II

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

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

image.png

2、解题思路

由于c++中包含了许多库的操作,但是在面试过程中,面试官可能不希望我们用太多的库去解题,特别是有些题目他的核心步骤就是写某个函数,所以,在自己能力范围之内,可以自己写某些函数。例如此题中的revrse函数。

3、代码

class Solution 
{
public:
    // reverse函数可以自己实现,也可以调用reverse库函数
    void reverse(string& s, int start, int end) 
    {
        for (int i = start, j = end; i < j; i++, j--) 
        {
            swap(s[i], s[j]);
        }
    }
    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, i, i + k - 1);
                continue;
            }
            // 3. 剩余字符少于 k 个,则将剩余字符全部反转。
            reverse(s, i, s.size() - 1);
        }
        return s;
    }
};