辗转相除法(GCD)求左旋转字符串
今天在牛客网上做了一道题,题意就是求左旋转字符串。我使用辗转相除法解之,一次性AC通过。实话说,每次写算法一次性通过,甚至一点编译错误都没有,我觉得这就是对我最好的嘉奖。
题目描述:
汇编语言中有一种移位指令叫做循环左移(ROL),现在有个简单的任务,就是用字符串模拟这个指令的运算结果。对于一个给定的字符序列S,请你把其循环左移K位后的序列输出。例如,字符序列S=”abcXYZdef”,要求输出循环左移3位后的结果,即“XYZdefabc”。是不是很简单?OK,搞定它!
这么多废话,实际上就是求左旋转字符串。这里我先给出我的辗转相除法代码:
class Solution {
public:
string LeftRotateString(string str, int n) {
int length = str.length();
if(length <= n)
return str;
char *sz = new char[length+1];
strcpy(sz, str.c_str());
int times = gcd(length, n);
while(times--) //注意这里先判断,再--,所以下面times已经是减过了的
rotate(sz+times, length, n);
string res(sz);
delete []sz;
return res;
}
void rotate(char* sz, int length, int n){
char* p1 = sz;
char* p2 = sz + n;
char tmp = *sz;
while(p2 != sz){
*p1 = *p2;
p1 = p2;
if(p2 + n > sz + length - 1)
p2 = sz + (n - ((sz + length) - p2));
else
p2 += n;
}
*p1 = tmp;
}
int gcd(int m, int n){
while(n != 0){
if(m < n)
std::swap(m, n);
int tmp = m % n;
m = n;
n = tmp;
}
return m;
}
};
思路大致是这样的:对于一个字符串,先求出GCD,也就是字符串长度和要旋转个数的最大公约数。这个最大公约数是我们即将要用到的循环链的数目。也就是执行rotate循环的次数。
现在解释什么是循环链。比如“1234”,我们求它把前面3个字符放到后面的情况,即n=3。此时GCD(4,3)=1,即times=1。那么分析rotate函数。由于在该函数中我们用tmp保存了sz,你需要用tmp保存了这个值,也就是p1的值,我们就可以使用*p2覆盖该位置的值了,并且p2=p1+n。对于该情况,rotate函数依照代码执行的流程为(用下划线表示空位,也就是p1指向的将被覆盖的位置):
_ 2 3 4 -> 4 2 3 _ -> 4 2 _ 3 -> 4 _ 2 3 -> 4 1 2 3(这是最后一步:*p1=tmp)
如上,这个rotate函数只需执行一次就可以达到目的,旋转3个字符要求达成。这就叫做一次循环链,最大公约数times的值就说明了这个问题。
再举一例,有两个循环链的例子:还是“1234”,求把前两个字符放在后面的情况,即n=2。GCD(4,2)=2,可知有两个循环链。我们来验证一下:
第一次:times=1(由于times--),所以p1=2,p2=4(p2=p1+n),所以循环流程:
1 _ 3 4 -> 1 4 3 _ -> 1 4 3 2
第二次:times=0,并且p1=1,p2=3,循环流程为:
_ 4 3 2 -> 3 4 _ 2 -> 3 4 1 2
没错,循环两次就成功解决问题了,所以最大公约数就是循环链的数目,也就是执行rotate函数的次数。
我在牛课网上看了,大多数人可能使用没有改变内存的substr,有的人使用三次reverse,没有见到有人使用GCD。我之前也用reverse,但是现在熟悉了新技能,那就用它吧。
对了,我的思路是以前剖析STL源代码学习的,如果感兴趣可以看STL rotate函数的实现,对不同迭代器重载不同的版本,其中RandomIterator版本使用的就是GCD。
posted on 2019-02-06 00:09 TestCNBlogss 阅读(269) 评论(0) 编辑 收藏 举报