1008. 陣列元素迴圈右移問題 (20)

一個數組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");
	}	
}