LintCode-8. 旋转字符串

题目描述

给定一个字符串和一个偏移量，根据偏移量旋转字符串(从左向右旋转)
样例
对于字符串 “abcdefg”.
offset=0 => “abcdefg”
offset=1 => “gabcdef”
offset=2 => “fgabcde”
offset=3 => “efgabcd”
挑战
在数组上原地旋转，使用O(1)的额外空间

分析

看完题目，我的想法是把整个字符串分成两个部分。根据偏移量的不同，两个部分的长度会相应变化。最后将后面的部分放到前面，前面的部分放到后面就可以了。这里还要考虑到偏移量大于字符串长度的情况。

代码

class Solution {
public:
    /*
     * @param str: An array of char
     * @param offset: An integer
     * @return: nothing
     */
    void rotateString(string &str, int offset) {
        // write your code here
        int l = str.length();
        if(l == 0) return;  //吸取上一道题（6.合并排序数组）的教训，事先将字符串为空的情况列出来
        offset = offset % l;  //考虑到偏移量大于字符串长度的情况
        if(offset > 0) {
            string s1 = str.substr(0, l-offset);
            string s2 = str.substr(l-offset, l-1);
            str.replace(0, offset-1, s2);
            str.replace(offset, l-1, s1);
        }

    }
};

优化

去网上看看有没有更好的方法，看到了一个关于这个问题的一个经典算法——三次反转。比如原字符串为“123456”，偏移量为3，则将”123”反转变成”321”,将”456”反转成”654”，最后再将整个字符串”321654”反转变成”456123”。其实和我的方法差不多，只不过我是利用了string库函数提取子串（substr）和字符串替换（replace）。

总结

string类的用法