力扣 541 反转字符串 II
题目描述
给定一个字符串 s 和一个整数 k,你需要对从字符串开头算起的每隔 2k 个字符的前 k 个字符进行反转。
如果剩余字符少于 k 个,则将剩余字符全部反转。
如果剩余字符小于 2k 但大于或等于 k 个,则反转前 k 个字符,其余字符保持原样。
示例:
输入: s = "abcdefg", k = 2
输出: "bacdfeg"
提示:
- 该字符串只包含小写英文字母。
- 给定字符串的长度和 k 在
[1, 10000]
范围内。
解决方案
暴力
算法和想法
我们直接翻转每个 2k 字符块。
每个块开始于 2k 的倍数,也就是 0, 2k, 4k, 6k, …。需要注意的一件是:如果没有足够的字符,我们并不需要翻转这个块。
为了翻转从 i 到 j 的字符块,我们可以交换位于 i++ 和 j– 的字符。
实现代码
Python 实现
class Solution(object):
def reverseStr(self, s, k):
a = list(s)
for i in xrange(0, len(a), 2*k):
a[i:i+k] = reversed(a[i:i+k])
return "".join(a)
Java 实现
class Solution {
public String reverseStr(String s, int k) {
char[] a = s.toCharArray();
for (int start = 0; start < a.length; start += 2 * k) {
int i = start, j = Math.min(start + k - 1, a.length - 1);
while (i < j) {
char tmp = a[i];
a[i++] = a[j];
a[j--] = tmp;
}
}
return new String(a);
}
}
复杂度分析
时间复杂度:O(N),其中 N 是 s 的大小。我们建立一个辅助数组,用来翻转 s 的一半字符。
空间复杂度:O(N),a的大小。
酷客网相关文章:
评论前必须登录!
注册