leetcode 541. 反转字符串 II 简单

圣保罗的大教堂2025-02-27 16:48

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

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

示例 1：

复制代码

输入：s = "abcdefg", k = 2
输出："bacdfeg"

示例 2：

复制代码

输入：s = "abcd", k = 2
输出："bacd"

提示：

1 <= s.length <= 10^4
s 仅由小写英文组成
1 <= k <= 10^4

分析：直接按题意进行模拟。反转每个下标从 2k 的倍数开始的，长度为 k 的子串。若该子串长度不足 k，则反转整个子串。

cpp 复制代码

char* reverseStr(char* s, int k) {
    int len=strlen(s),f=1,index=0;
    char *ans=(char*)malloc(sizeof(char)*(len+1));
    ans[len]=0;
    for(int i=0;i<len;i=i+k*2)
    {
        if(i+k>=len)
        {
            for(int j=len-1;j>=i;--j)
                ans[index++]=s[j];
        }
        else if(i+2*k>=len)
        {
            for(int j=i+k-1;j>=i;--j)
                ans[index++]=s[j];
            for(int j=i+k;j<len;++j)
                ans[index++]=s[j];
        }
        else
        {
            for(int j=i+k-1,t=0;t<k;--j,++t)
                ans[index++]=s[j];
            for(int j=i+k,t=0;t<k;++j,++t)
                ans[index++]=s[j];
        }
    }
    return ans;
}