LeetCode 541.反转字符串||

题目描述

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

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

示例 1：

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

示例 2：

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

提示：

• 1 <= s.length <= 10^4
• s 仅由小写英文组成
• 1 <= k <= 10^4

思路

本题可以用模拟来解决，反转可以直接用库函数。注意要让i每次移动2*k个位置，从而减少遍历的次数。

代码

C++版：

class Solution {
public:
    string reverseStr(string s, int k) {
        // 每次移动2*k
        for(int i=0;i<s.size();i+=2*k){
            if(i+k<=s.size()){
                // 剩余字符小于 2k 但大于或等于 k 个，则反转前 k 个字符
                reverse(s.begin()+i,s.begin()+i+k);
            }else{
                // 剩余字符少于 k 个，则将剩余字符全部反转
                reverse(s.begin()+i,s.end());
            }
        }
        return s;
    }
};

Python版：

class Solution:
    def reverseStr(self, s: str, k: int) -> str:
        def reverse_substring(text):
            left, right = 0, len(text) - 1
            while left < right:
                text[left], text[right] = text[right], text[left]
                left += 1
                right -= 1
            return text
        
        res = list(s)

        for cur in range(0, len(s), 2 * k):
            res[cur: cur + k] = reverse_substring(res[cur: cur + k])
        
        return ''.join(res)

需要注意的地方

1.当需要固定规律一段一段去处理字符串的时候，要想办法在for循环上做文章。