传送门
https://www.patest.cn/contests/pat-b-practise/1008
题目
一个数组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
分析
可以利用栈或者队列都可以实现,我选用的是栈,先根据移动次数,将数字压入栈,然后将数组循环赋值,移动至数组尾部,然后从栈顶弹出元素,在依次从数组头部放入,即可。
注意,移动次数要对数组长度取余。另外还要考虑移动次数0的情况,因为0不能作为除数。
源代码
//C/C++实现
#include <stdio.h>
#include <iostream>
#include <vector>
#include <stack>
using namespace std;
int main(){
int n, m, rest, tmp;
scanf("%d%d", &n, &m);
vector<int> v(n);
for(int i=0;i<n;i++){
scanf("%d", &v[i]);
}
if(m != 0 && (m % n != 0)){
tmp = rest = m % n;
stack<int> s;
for(int j=n-1;j>=0;j--){
if(tmp != 0){
s.push(v[j]);
tmp --;
}
else{
v[j+rest] = v[j];
}
}
for(int k=0;k<rest;k++){
v[k] = s.top();
s.pop();
}
}
for(int l=0;l<n;l++){
printf("%d", v[l]);
if(l != n-1){
printf("%c", ' ');
}
else{
printf("%c", '\n');
}
}
return 0;
}
网友评论