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

传送门

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;
}

当前网速较慢或者你使用的浏览器不支持博客特定功能,请尝试刷新或换用Chrome、Firefox等现代浏览器