题目
给定一个字符串 s 和一个整数 k,从字符串开头算起,每 2k 个字符反转前 k 个字符。
如果剩余字符少于 k 个,则将剩余字符全部反转。
如果剩余字符小于 2k 但大于或等于 k 个,则反转前 k 个字符,其余字符保持原样。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/reverse-string-ii
思路
- 反转字符串 老生常谈 用栈 或者双指针 一个向左一个向右
- 这道题用人话讲 就是每隔k个反转k个
- 问题在于第一次反转后的处理
- 我们用双指针 头指针在0,尾指针在K + 0
分三种情况
第一种:如果剩余大于2k个 那么头指针 = k + k 尾指针 = k+ k + k
第二种:如果剩余大于k个 小于2k个 那么头指针= k+k 尾指针 = 字符串长度
第三种:如果剩余小于k个 那么直接返回 不用反转了
代码实现(可优化)
public String reverseStr(String s, int k) {
// 如果 = 1 不用反转
if (k<=1) {
return s;
}
char[] chars = s.toCharArray();
// 头指针
int a = 0;
// 尾指针
int b = k - 1;
// 记录反转到哪
int c = k - 1;
int length = chars.length;
if (length <= k) {
c = b = length - 1;
}
while (true) {
if (b <= a ) {
// 三种情况判断
int i = length - c;
if (i < k) {
break;
} else if (i > k && i < 2 * k) {
b = length - 1;
} else if (i> 2 * k) {
b = c + 2 * k;
}
a = c + k + 1;
c = b;
}else {
// 反转处理
char aChar = chars[a];
char bChar = chars[b];
chars[a] = bChar;
chars[b] = aChar;
b--;
a++;
}
}
return new String(chars);
}