题目描述
拥有n个整数的数组a,下标为0,1···n-1,现将整个数组整体右移k位,求移动后的数组,以函数实现
要求
请尽量使用最小的额外空间和算法复杂度
示例
a={1,2,3,4,5,6,7},右移3位,a={5,6,7,1,2,3,4}
解题思路
- 将{1,2,3,4}翻转
a={4,3,2,1,5,6,7} - 将{5,6,7}翻转
a={4,3,2,1,7,6,5} - 将{4,3,2,1,7,6,5}翻转
a={5,6,7,1,2,3,4}
步骤1.2.3的顺序无所谓。
void f(int* a, int size, int k) {
k %= size;
for (int i = 0; i < (size - k) / 2; i++)
swap(a[i], a[size - k - 1 - i]);
//将左边部分翻转
for (int i = 0; i < k / 2; i++)
swap(a[size - k + i], a[size - 1 - i]);
//将右边部分翻转
for (int i = 0; i < size / 2; i++)
swap(a[i], a[size - 1 - i]);
//将整个数组翻转
}