题目描述
给定一个字符串和一个偏移量,根据偏移量旋转字符串(从左向右旋转)
样例
对于字符串 “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)。