给定一个字符串 s 和一个整数 k,从字符串开头算起, 每计数至 2k 个字符,就反转这 2k 个字符中的前 k 个字符。如果剩余字符少于 k 个,则将剩余字符全部反转。如果剩余字符小于 2k 但大于或等于 k 个,则反转前 k 个字符,其余字符保持原样。
示例:
输入:s = "abcdefg",k = 2
输出:"bacdfeg"
一开始看这个题目可能会有点混乱,但归根结底就是字符串的反转,只不过这道题目设置了多个条件,我们结合示例一个一个条件看就很简单了。
如示例中,整数为2,那根据要求,字符串中每4个字符为一组,每组的前两个字符反转。如果最后的字符凑不够4个,就又分了两种情况,剩余字符数小于2的话就全部反转,剩余字符数大于等于2小于4时就反转前两个字符。(其实你换个角度想,归根结底就一个条件,就是4个字符为一组,每组的前两个字符反转)
完整代码如下:
class Solution:
def reverseStr(self, s: str, k: int) -> str:
# Two pointers. Another is inside the loop.
p = 0
while p < len(s):
p2 = p + k
# Written in this could be more pythonic.
s = s[:p] + s[p: p2][::-1] + s[p2:]
p = p + 2 * k
return s
代码其实很简单,但需要你去理解。首先,定义一个p赋值为0,while循环的要求是p小于字符串长度。在循环中再次定义一个节点为p2,将其p+k赋值给它。
接着的关键代码,s = s[:p] + s[p: p2][::-1] + s[p2:]。相信很多基础不行的同学在循环中定义条件时,喜欢将第一次循环代入进去来判断自己定义的条件对不对。在本题中,将一开始的p=0代入,那么p2=2,s=s[:0]+s[0:2][::-1]+s[2:],代码很正确,条件也符合题目。但有些同学会疑惑了,为什么要加一个s[:0]?这不是多此一举?我直接让s =s[p: p2][::-1] + s[p2:],前面k个字符反转,后面k个不变。
在第一次循环中,确实s[:0]的作用并不明显,但你放到第二个循环中看,s=s[:4]+s[4:6]+s[6:],你如果不加s[:p],是不是当前循环所在组的前面字符都被省略了?因此s = s[:p] + s[p: p2][::-1] + s[p2:]中s[:p]是为了保证当前组之前的字符不被修改或忽略,s[p: p2][::-1]是为了在当前组进行反转操作,s[p2:]是为了当前组之后的字符不被修改或忽略。