541. 反转字符串 II

541. 反转字符串 II

LeetCode题目要求

给定一个字符串 s 和一个整数 k，从字符串开头算起，每计数至 2k 个字符，就反转这 2k 字符中的前 k 个字符。
如果剩余字符少于 k 个，则将剩余字符全部反转。
如果剩余字符小于 2k 但大于或等于 k 个，则反转前 k 个字符，其余字符保持原样。

示例

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

解题思路

反转规则：
// s = "abcdefg", k = 2
// 从头计数，2k=4 到d，反转前 k=2 个，即反转 ab 为 ba，成为bacd
// 此时剩余字符串为 feg, 小于 2k=4, 但是大于 k=2 个，那么反转前 k=2 个，即 efg -> feg
// 另外一种情况是 如果 剩余字符串 小于 k，就全部反转
// 最终的结果就是两个反转的字符串的结果为 bacdfeg"

    // 每计数至 2k 个字符，就反转这 2k 字符中的前 k 个字符
    // 如果剩余字符少于 k 个，则将剩余字符全部反转
    // 如果剩余字符小于 2k 但大于或等于 k 个，则反转前 k 个字符，其余字符保持原样

总结起来就是步进 2k 后，如果 k + i 在字符范围内，那就直接将他们前 k 个反转；其余剩下的就都反转
本题在反转过程中因为是部分反转，使用 char 数组来处理相对比直接反转 String 要简单一些

上代码

class Solution {

    public String reverseStr(String s, int k) {

        char[] cs = s.toCharArray();

        for (int i = 0; i < cs.length; i += k*2){
            if (i + k <= cs.length) {
                reverse(cs, i, i+k-1);
                continue;
            }
            reverse(cs, i, s.length()-1);
        }
        return new String(cs);        
    }

    public void reverse(char[] cs, int i, int j) {
        for (; i < j; i++, j--) {
            char temp  = cs[i];
            cs[i] = cs[j];
            cs[j] = temp;
        }
    }
}

重难点

附：学习资料链接