给定一个字符串 s 和一个整数 k,从字符串开头算起,每计数至 2k 个字符,就反转这 2k 字符中的前 k 个字符。
如果剩余字符少于 k 个,则将剩余字符全部反转。
如果剩余字符小于 2k 但大于或等于 k 个,则反转前 k 个字符,其余字符保持原样。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/reverse-string-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
import java.util.Scanner;
class Solution {
private void swap(char[] str, int a, int b) {
char t = str[a];
str[a] = str[b];
str[b] = t;
}
private void reverse(char[] str, int left, int right) {
while (left < right) {
swap(str, left++, right--);
}
}
public String reverseStr(String s, int k) {
char[] str = s.toCharArray();
int index = 0;
while (index < s.length()) {
reverse(str, index, Math.min(index + k - 1, s.length() - 1));
index += 2 * k;
}
return new String(str);
}
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
while (in.hasNext()) {
System.out.println(new Solution().reverseStr(in.next(), in.nextInt()));
}
}
}