PAT-乙-1008 1008 数组元素循环右移问题 (20 分)
代码
#include <stdio.h>
void reverse(int a[], int start, int end){
for(int i=start; i<=(start+end)/2; i++){
int tmp = a[i];
a[i] = a[start+end-i];
a[start+end-i] = tmp;
}
}
int main(){
int n, m;
scanf("%d %d", &n, &m);
m %= n;
int a[n];
for(int i=0; i<n; i++){
scanf("%d", &a[i]);
}
reverse(a, 0, n-m-1);
reverse(a, n-m, n-1);
reverse(a, 0, n-1);
printf("%d", a[0]);
for(int i=1; i<n; i++){
printf(" %d", a[i]);
}
printf("\n");
return 0;
}
注解
(法一),不移动元素,改变printf的输出元素顺序即可。是偷懒的做法。
(法二),设一个额外的数组。增加了空间复杂度,但不符合题目要求,有点是代码简单。
(法三),老老实实按照题目所说,循环右移。严蔚敏老师《数据结构》链表章节中,有对此解法的介绍。
先反转前面部分,再反转后面部分,最后再对整体进行反转。三步走策略。
也就是代码中这三行,关键是数字的设定,哪里该减一,哪里不能减一,方法就是用具体数字代入试一试。另外,reverse函数的编写,也是同样,重点是下标的计算,功能是原地反转。
reverse(a, 0, n-m-1);
reverse(a, n-m, n-1);
reverse(a, 0, n-1);
此外,此题有一个坑点,M有可能大于等于N。
因此m = m%n;这句代码是不可缺少的,否则case 1,case 2错误,扣6分。