题目
给定一个字符串 s 和一个整数 k，从字符串开头算起，每 2k 个字符反转前 k 个字符。
- 如果剩余字符少于 k 个，则将剩余字符全部反转。
- 如果剩余字符小于 2k 但大于或等于 k 个，则反转前 k 个字符，其余字符保持原样。
示例 1：
输入：s = "abcdefg", k = 2 输出："bacdfeg"
示例 2：
输入：s = "abcd", k = 2 输出："bacd"
解题
方法一：
python解法
class Solution: def reverseStr(self, s: str, k: int) -> str: res = "" for i in range(len(s[::k])): if i%2==0: tmp = s[i*k : i*k+k][::-1] res+=tmp else: tmp = s[i*k : i*k+k] res+=tmp return res
由于python中 s[a:b] 其中b可以超过索引，s[a:b]取到的就是a到最后的一个索引 （所以这道题这么写没有问题）
但是s[b]，中b不能超过索引。
不想让它超过索引，也可以使用这个写法
class Solution: def reverseStr(self, s: str, k: int) -> str: n = len(s) res = "" for i in range(len(s[::k])): if i%2==0: tmp = s[i*k : i*k+k][::-1] if i*k+k<n else s[i*k : n][::-1] res+=tmp else: tmp = s[i*k : i*k+k] if i*k<n else s[i*k : n] res+=tmp return res
方法二：
python解法
class Solution: def reverseStr(self, s: str, k: int) -> str: t = list(s) for i in range(0, len(t), 2 * k): t[i: i + k] = reversed(t[i: i + k]) return "".join(t)
c++解法
class Solution { public: string reverseStr(string s, int k) { int n=s.size(); for(int i=0;i<n;i+=2*k){ reverse(s.begin()+i,s.begin()+min(i+k,n)); } return s; } };
或者这样写逻辑更加清晰
class Solution { public: string reverseStr(string s, int k) { int n=s.size(); for(int i=0;i<n;i+=2*k){ if(n-i<k){ reverse(s.begin()+i,s.end()); } else{ reverse(s.begin()+i,s.begin()+i+k); } } return s; } };
java
class Solution { void swap(char[] s,int l,int r){ for(int i=l,j=r;i<j;i++,j--){ char tmp=s[i]; s[i]=s[j]; s[j]=tmp; } } public String reverseStr(String s, int k) { char[] str=s.toCharArray(); for(int i=0;i<str.length;i+=2*k){ swap(str,i,Math.min(i+k-1,str.length-1)); } return String.valueOf(str); } }
