代码随想录二刷 |字符串 |反转字符串II
- 题目描述
- 解题思路 & 代码实现
题目描述
541.反转字符串II
给定一个字符串 s 和一个整数 k,从字符串开头算起,每计数至 2k 个字符,就反转这 2k 字符中的前 k 个字符。
如果剩余字符少于 k 个,则将剩余字符全部反转。
如果剩余字符小于 2k 但大于或等于 k 个,则反转前 k 个字符,其余字符保持原样。
示例 1:
输入:s = “abcdefg”, k = 2
输出:“bacdfeg”
示例 2:
输入:s = “abcd”, k = 2
输出:“bacd”
提示:
- 1 <= s.length <= 104
- s 仅由小写英文组成
- 1 <= k <= 104
解题思路 & 代码实现
再循环中让i
的步长为2k
即可,剩余字符 = s.szie() - i
,根据条件可得不等式:k <= s.size() - i < 2k
,化简一下可得k + i <= s.size() < 2k + i
,我们要找的是最小满足条件,因此用k + i <= s.szie()
就可以。
class Solution {
public:string reverseStr(string s, int k) {for (int i = 0; i < s.size(); i += (2 * k)) {// 1.每隔 2k 个字符的前 k 个字符进行反转// 2.剩余字符小于 2k 且大于等于 k ,反转前 k 个字符if (i + k <= s.size()) {reverse(s.begin() + i, s.begin() + i + k);} else { // 3.剩余字符小于 k 个,则将剩余字符全部反转reverse(s.begin() + i, s.end());}}return s;}
};