1008 数组元素循环右移问题 (20)(20 分)
一个数组A中存有N(N>0)个整数,在不允许使用另外数组的前提下,将每个整数循环向右移M(M>=0)个位置,即将A中的数据由(A~0~ A~1~……A~N-1~)变换为(A~N-M~ …… A~N-1~ A~0~ A~1~……A~N-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
O(n)时间复杂度的答案如下:
#include <iostream>
#include <string>
#include <cmath>
using namespace std;
int main()
{
int N, M;
cin >> N >> M;
int A[N];
for (int i = 0; i < N; i ++)
{
cin >> A[i];
}
int temp, c = 0,p1 = 0, p2 = (p1 + M) % N;
while (c < N - 1)
{
temp = A[p1];
A[p1] = A[p2];
A[p2] = temp;
p2 = (p2 + M) % N;
if (p1 == p2)
{
p1 = (p1 + 1) % N;
p2 = (p1 + M) % N;
c ++;
}
c ++;
}
for (int i = 0; i < N; i ++)
{
cout << A[i];
if(i < N-1)
{
cout << " ";
}
}
return 0;
}