leetcode.541.反转字符串 II

541. 反转字符串 II

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

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

示例 1：

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

示例 2：

输入：s = “abcd”, k = 2
输出：“bacd”

题意理解：
遍历字符串，每遍历2K个字符就将前K个字符反序。

• 注意最后剩余字符的个数
• 使用双指针确定反序子字符串的下标位置，使用reverse函数进行反转或自己写reverse函数

1. C++

class Solution {
public:
    // string reverseStr(string s, int k) {
    //     int n = s.size();
    //     int i=0;
    //     for(; i
    //         reverse_l(s, i*2*k, (i)*2*k+k-1);
    //     }
    //     if (n-i*2*k > k){
    //         reverse_l(s, i*2*k, (i)*2*k+k-1);
    //     }
    //     else{
    //         reverse_l(s, i*2*k, n-1);
    //     }
    //     return s;
    // }
    string reverseStr(string s, int k) {
        int n = s.size();
        for(int i=0; i<n; i+=2*k){
            reverse_l(s, i, min(n-1, i+k-1));
        }
        return s;
    }
    void reverse_l(string &s, int i, int j){
        while(i<j){
            swap(s[i],s[j]);
            i++;
            j--;
        }
    }
};

2. python

class Solution:
    def reverseStr(self, s: str, k: int) -> str:
        t = list(s)
        for i in range(0, len(t), 2 * k):
            t[i: i + k] = reversed(t[i: i + k])
        return "".join(t)