1、题目:设计一个算法,把一个含有N个元素的数组循环右移K位,要求时间复杂度为O(N),且只允许使用两个附加变量。假如数组为abcd1234,循环右移4位的话,即1234abcd。不妨设k是一个非负的整数,当k为负整数的时候,右移k位,相当于左移-k位。
可以发现,每个元素右移N位后都会回到自己的位置上。因此,如果K>N,右移K-N之后的数组序列根右移K位的结果是一样的。即有规律:右移K位之后的情景,跟右移K‘=K%N位之后的情形一样。
RightShift(int* arr, int N, int K)
{
K %= N;
while(K--)
{
int t = arr[N-1];
for(int i=N-1; i>0; i--)
arr[i] = arr[i-1];
arr[0] = t;
}
}
解法二:
以abcd1234为例,即使右移4位变成了1234abcd,其中有两段的顺序是不变的:1234和abcd,看成两个整体,右移K位的过程就是把数组的两部分交换一下。
Reverse(int* arr, int b, int e)
{
for(; b<e; b++, e--)
{
int temp = arr[e];
arr[e] = arr[b];
arr[b] = temp;
}
}
RightShift(int* arr, int N, int K)
{
K %= N;
Reverse(arr, 0, N-K-1);
Reverse(arr, N-K, N-1);
Reverse(arr, 0, N-1);
}