1008. 数组元素循环右移问题 (20)--PAT乙级

1008. 数组元素循环右移问题 (20)

时间限制

400 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

一个数组A中存有N（N>0）个整数，在不允许使用另外数组的前提下，将每个整数循环向右移M（M>=0）个位置，即将A中的数据由（A₀A₁……A_N-1）变换为（A_N-M …… A_N-1 A₀ A₁……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

分析：简单的下标变换，注意这里的M可能大于N需要取模。这里用a[0]到a[N-1]存储原数组A，用a[N]到a[2*N-1]

存储右移后的数组。下标i+N对应原数组的下标为（i+N-M%N)%N）。

代码如下：

#include<cstdio>
#include<string>
#include<cstring>
const int MAXN = 200 + 5; 
using namespace std;

int N,M,a[MAXN];
int main() 
{
	while(scanf("%d%d", &N, &M) != EOF)
	{
		memset(a, 0, sizeof(a));//初始化 
		for(int i = 0; i < N; i++)
		scanf("%d", &a[i]);
		for(int i = 0; i < N; i++)
		{
			int n = (i + N - M % N) % N;
			a[i + N] = a[n];//i+N所对应的平移前的下标为n 
		}
		for(int i = N; i < 2 * N; i++)//下标N~2 * N - 1对应平移后的结果 
		{
			printf("%d", a[i]);
			if(i != 2 * N - 1)
				printf(" ");
		}
		printf("\n");
	}	
}

编辑于2017/09/13

法二：三次原地倒置

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>

const int MAXN = 200 + 5; 
using namespace std;

int N,M,a[MAXN];
int main() 
{
	int index;
	while(scanf("%d%d", &N, &M) != EOF)
	{
		memset(a, 0, sizeof(a));//初始化 
		for(int i = 0; i < N; i++)
			scanf("%d", &a[i]);
		index = N - M % N;
		//倒置三次 
		reverse(a, a + index);
		reverse(a + index, a + N);
		reverse(a, a + N);
		
		for(int i = 0; i < N; ++i)//输出 
		{
			if(i != N - 1)
				printf("%d ", a[i]);
			else printf("%d", a[i]);
		}
		printf("\n");
	}	
}