LeetCode 541 反转字符串II

题目

给定一个字符串 s 和一个整数 k，你需要对从字符串开头算起的每隔 2k 个字符的前 k 个字符进行反转。

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

示例:

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

提示：

• 该字符串只包含小写英文字母。
• 给定字符串的长度和 k 在 [1, 10000] 范围内。

来源：力扣（LeetCode）

链接：https://leetcode-cn.com/problems/reverse-string-ii

著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

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

class Solution {
    public String reverseStr(String s, int k) {
        StringBuilder sb = new StringBuilder();
        int len = s.length();
        for(int i = 0; i < len; i += 2 * k) {
            StringBuilder tmp = new StringBuilder();
            // 剩余字符大于 k 个，每隔 2k 个字符的前 k 个字符进行反转
            if(i + k <= len){
                // 反转前 k 个字符
                tmp.append(s.substring(i, i + k));
                sb.append(tmp.reverse());
                // 反转完前 k 个字符之后，如果紧接着还有 k 个字符，则直接加入这 k 个字符
                if(i + 2 * k <= len) {
                    sb.append(s.substring(i + k ,i + 2 * k));
                } else { // 不足 k 个字符，则直接加入剩下所有字符
                    sb.append(s.substring(i + k,len));
                }
                continue;
            }
            // 剩余字符少于 k 个，则将剩余字符全部反转
            tmp.append(s.substring(i,len));
            sb.append(tmp.reverse());
        }
        return sb.toString();
    }
}

时间复杂度：O(n)
空间复杂度：O(n)