PAT 1008 数组元素循环右移问题

落日映苍穹つ 2022-08-18 02:37 48阅读 0赞

一个数组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的情况

#include <iostream>
    #include <vector>
    #include <algorithm>
    #include <cstdio>
    #include <cmath>
    #include <cstdlib>
    #include <cstring>
    
    using namespace std;
    
    /* 进行收尾的反转 */
    void reserve(vector<int> &vec, const int begins, const int ends)
    {
        int i = 0, tmp = 0;;
        int len = (begins + ends) / 2;
        int sum = begins + ends;
    
        if (ends <= 0 || begins >= vec.size() - 1)
        {
            return;
        }
    
        for(i = begins; i <= len; i++)
        {
            tmp = vec[i];
            vec[i] = vec[sum - i];
            vec[sum - i] = tmp;
        }
    }
    
    int main()
    {
        int N = 0, M = 0, i = 0;
        int data = 0;
        vector<int> vecNUm;
    
        cin >> N >> M;
    
        /* 可能出现M > N */
        M = M % N;
    
        /* 输入N个数据 */
        for(i = 0; i < N; i++)
        {
            cin >> data;
            vecNUm.push_back(data);
        }
    
        /* 先整体反转 */
        reserve(vecNUm,0,vecNUm.size()-1);
    
        /* 前半部分反转 */
        reserve(vecNUm,0,M-1);
    
        /* 后半部分反转 */
        reserve(vecNUm,M,vecNUm.size()-1);
    
        for(i = 0; i < N; i++)
        {
    
            if (i == 0)
            {
                cout << vecNUm[i];
            }
            else
            {
                cout << " " << vecNUm[i];
            }
        }
        cout << endl;
    }