题目描述
一个数组A中存有N(N>0)个整数,在不允许使用另外数组的前提下,将每个整数循环向右移M(M>=0)个位置,即将A中的数据由(A0 A1……AN-1)变换为(AN-M …… AN-1 A0 A1……AN-M-1)(最后M个数循环移至最前面的M个位置)。如果需要考虑程序移动数据的次数尽量少,要如何设计移动的方法?
输入描述:
每个输入包含一个测试用例,第1行输入N ( 1<=N<=100)、M(M>=0);第2行输入N个整数,之间用空格分隔。
输出描述:
在一行中输出循环右移M位以后的整数序列,之间用空格分隔,序列结尾不能有多余空格。
输入例子:
6 2
1 2 3 4 5 6
输出例子:
5 6 1 2 3 4
1 int size,count; 2 cin >> size >> count; 3 int a[200] ={0}; 4 int nums = count%size; 5 for(int i=0;i<size;++i) 6 { 7 cin >> a[i]; 8 if(nums == size) 9 nums = 0; 10 a[100+(nums++)] = a[i]; 11 } 12 for(int i=0;i<size;++i) 13 cout << a[i+100] << ' ';
之前这样写的,直接定义一个200的数组,把数组后移到后半边的数组里。但是还是借助了之外的内存。
由于有上边红字限制。在一个数组里边移动元素,势必会覆盖掉当前位置的元素,借助于其他内存保存这些即将被覆盖的元素又是不合法的。所以这里我们需要另辟蹊径,采用三次reverse操作,实现“移动”,时间空间复杂度都不超过O(n)。
原理如图所示:
数组的倒置操作也很简单,头尾指针指向的值互换,直至两个指针相交。
AC代码:
#include <iostream> using namespace std; class Solutions{ public: void reverse(int arr[],int prev,int back) { int tmp; while(prev<back) { tmp = arr[back]; arr[back] = arr[prev]; arr[prev] = tmp; ++prev; --back; } } void print(int arr[],int prev,int back) { for(int i=prev;i<back;++i) { cout << arr[i]; if(i!=back-1) cout << ' '; } } }; int main() { Solutions s; int size,number; cin >> size >> number; int a[100]; for(int i=0;i<size;++i) cin >> a[i]; number = number%size; s.reverse(a,0,size-1-number); s.reverse(a,size-number,size-1); s.reverse(a,0,size-1); s.print(a,0,size); }