LeetCode 541. 反转字符串 II

题目链接:LeetCode 541. 反转字符串 II

题意:

给定一个字符串 s 和一个整数 k,从字符串开头算起,每计数至 2k 个字符,就反转这 2k 字符中的前 k 个字符。

  • 如果剩余字符少于 k 个,则将剩余字符全部反转。
  • 如果剩余字符小于 2k 但大于或等于 k 个,则反转前 k 个字符,其余字符保持原样。

解题思路:

从前往后遍历字符串s,下标 i 每次移动2k,然后判断是否需要有反转的区间。

完整代码如下:

func reverseStr(s string, k int) string {
    ss:=[]byte(s)
    for i:=0;i<n;i += 2*k{
         if i + k <= n {  //剩余字符大于 k 个,则反转 k 个
               for a,b:=i,i+k-1;a<b;a,b=a+1,b-1{
                    ss[a],ss[b] = ss[b],ss[a]
                }
        } else {
                // 3. 剩余字符少于 k 个,则将剩余字符全部反转。
                for a,b:=i,n-1;a<b;a,b=a+1,b-1{
                    ss[a],ss[b] = ss[b],ss[a]
                }
        }
    }
    return string(ss)
}
posted @ 2023-05-10 14:58  小星code  阅读(12)  评论(0)    收藏  举报