今天朋友碰到一个面试题:
设计一个算法,把一个含有N个元素的数组循环右移K位,要求时间复杂度位O(N),空间复杂度为O(1)。
顺理成章的思考方式就是挨个换位,
123456789
789123456 // 如果N是(K%N)的倍数,则需要多个起点
567891234 // 设置一个起点就可以交换完
代码如下(声明只是思路,代码没有经过测试):
void move( type arr[], int N, int K )
{
int nOffset = K % N;
if ( nOffset == 0 )
return ;
int n = 0;
int i = 0;
int j = i;
type Temp1 = arr[j];
type Temp2 = 0;
do
{
j = ( j + nOffset ) % N;
Temp2 = arr[j];
arr[j] = Temp1;
// 如果N是K的倍数 则需要新设置跳跃的起点
if ( j == i )
{
j = ++i;
Temp1 = arr[j];
}
else
Temp1 = Temp2;
n++;
} while ( n < N -1 ) // 总共要交换(N-1)次
}