给定一个字符串 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
Code:
class Solution {
public:
string reverseStr(string s, int k) {
if(2*k>s.length())
{
if(k>=s.length())
{
reverse(s.begin(),s.end());
}
else
reverse(s.begin(),s.begin()+k);
return s;
}
else
{
string res="";
while(1)
{
string sub=s.substr(0,k);
reverse(sub.begin(),sub.end());
int leftlen=s.length()-2*k;
if(leftlen>=k&&leftlen<=2*k)
{
string sub2=s.substr(k,k);
string sub3=s.substr(2*k);
res+=sub;
res+=sub2;
reverse(sub3.begin(),sub3.begin()+k);
res+=sub3;
return res;
}
if(leftlen<k)
{
string sub2=s.substr(k,k);
string sub3=s.substr(2*k);
res+=sub;
res+=sub2;
reverse(sub3.begin(),sub3.end());
res+=sub3;
return res;
}
else
{
res+=sub;
res+=s.substr(k,k);
s=s.substr(2*k);
}
}
}
return "";
}
};