1008. 陣列元素迴圈右移問題 (20)--PAT乙級
阿新 • • 發佈:2019-01-25
1008. 陣列元素迴圈右移問題 (20)
時間限制 400 ms記憶體限制 65536 kB
程式碼長度限制 8000 B
判題程式 Standard
一個數組A中存有N(N>0)個整數,在不允許使用另外陣列的前提下,將每個整數迴圈向右移M(M>=0)個位置,即將A中的資料由(A0A1……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
分析:簡單的下標變換,注意這裡的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"); } }