leetcode541

题目:

给定一个字符串和一个整数 k,你需要对从字符串开头算起的每个 2k 个字符的前k个字符进行反转。如果剩余少于 k 个字符,则将剩余的所有全部反转。如果有小于 2k 但大于或等于 k 个字符,则反转前 k 个字符,并将剩余的字符保持原样。

示例:

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

要求:

  1. 该字符串只包含小写的英文字母。
  2. 给定字符串的长度和 k 在[1, 10000]范围内。

解决方案:

直接对给定字符串s中的1,3,5,7...个长度为k的小字符串进行反转即可。

解决代码:

  

class Solution {
public:
    string reverseStr(string s, int k) {
        int len = s.size();
        int count = len / k;
        for(int i=0;i<=count;i++){
            if(i%2 == 0){
                if((i+1)*k > len){
                    reverse(s.begin()+i*k,s.end());
                }else{
                    reverse(s.begin()+i*k,s.begin()+(i+1)*k);
                }
            }
        }
        return s;
    }
};

相关知识和程序bug:

1.reverse(first,last)函数:

函数功能为反转范围里的元素(reverse range),两个参数均为迭代器。

2.如果len<k:

此处代码为i<=count,这样可能会超出s的范围。但如果代码是i<count,则当len<k时,不会执行循环,从而会返回原字符串,导致错误。

 

 
  

猜你喜欢

转载自www.cnblogs.com/yxlsblog/p/10352911.html