编程之美中的题目:
设计一个算法,把一个含有N个元素的数组循环右移K位,要求时间复杂度为O(N),且只允许使用两个附加变量。
程序思路:
假设原始数组序列为abcd1234,循环右移4位。变换过程如下:
1. 逆序排列abcd: abcd1234 -> dcba1234
2. 逆序排列1234:dcba1234 -> dcba4321
3. 全部逆序:dcba4321 -> 1234abcd
注意:
1.循环右移K位相当于循环右移K%N位
2.K有可能为负数
3.写代码时注意异常情况判断
4.数组元素类型需要看题目要求,如果在面试中可以和面试官沟通一下,看是什么类型
5.左移和右移的解题思路是一样的
具体代码如下:
void Reverse(char *arr, int low, int high)
{
if(arr == NULL || low < 0 || high < 0 || low >= high)
return ;
while(low < high)
{
int temp = arr[high];
arr[high] = arr[low];
arr[low] = temp;
low++;
high--;
}
}
void RightShift(char *arr, int N, int K)
{
if(arr == NULL || N <= 0)
return ;
if(K < 0) //当K小于0时,右移K位相当于左移N-K位
K = N - (-K) % N;
else
K %= N; //右移K位相当于右移K%N位
Reverse(arr, 0, N - K - 1);
Reverse(arr, N - K, N - 1);
Reverse(arr, 0, N - 1);
}