1008 数组元素循环右移问题

原创于 2022-11-06 21:20:53 发布 · 282 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #c++ #开发语言

pat考试专栏收录该内容

27 篇文章

订阅专栏

本文介绍了一种在不使用额外数组的情况下，将数组中的元素向右循环移动M个位置的算法。该方法通过三次反转数组的不同部分来实现高效的数据移动。

一个数组A中存有N（>0）个整数，在不允许使用另外数组的前提下，将每个整数循环向右移M（≥0）个位置，即将A中的数据由（A0A1⋯AN−1）变换为（AN−M⋯AN−1A0A1⋯AN−M−1）（最后M个数循环移至最前面的M个位置）。如果需要考虑程序移动数据的次数尽量少，要如何设计移动的方法？

输入格式:

每个输入包含一个测试用例，第1行输入N（1≤N≤100）和M（≥0）；第2行输入N个整数，之间用空格分隔。

输出格式:

在一行中输出循环右移M位以后的整数序列，之间用空格分隔，序列结尾不能有多余空格。

输入样例:

6 2
1 2 3 4 5 6

输出样例:

5 6 1 2 3 4

思路：

先将整个数组a倒置，再将数组前m位倒置，最后将数组后n—m位倒置，此处使用了c++的reverse函数。

代码实现：

#include <bits/stdc++.h>
using namespace std;
int main() {
    int n, m;
    cin >> n >> m;
    vector<int> a(n);
    for (int i = 0; i < n; i++)
        cin >> a[i];
    m %= n;
    if (m != 0) {
        reverse(a.begin(), a.end());
        reverse(begin(a), begin(a) + m);
        reverse(begin(a) + m, begin(a) + n);
    }
    for (int i = 0; i < n - 1; i++)
        cout << a[i] << " ";
    cout << a[n - 1];
    return 0;
}