1008 数组元素循环右移问题 (20分)

PAT Based Level

Posted by Kenshin on June 7, 2020
本文字符数:英文613字 | 中文1541字

一个数组 A 中存有 N(>0)个整数,在不允许使用另外数组的前提下,将每个整数循环向右移 M(≥0)个位置,即将 A 中的数据由(A~0~A~1​~⋯A~​N−1~​​ )变换为(A~​N−M~​​ ⋯A~​N−1~​​ A~​0~​​ A~​1~​​ ⋯A~​N−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

我的代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
#include <iostream>
#include <vector>
using namespace std;
int main() {
    int n, m;
    cin >> n >> m;
    m %= n;
    vector<int> a(n, 0), b(0);
    for (int i = 0; i < n; i++)
        cin >> a[i];
    for (int i = n - 1; i >= 0; i--) {
        if (n - i - 1 < m)
            b.push_back(a[i]);
        if (i - m >= 0)
            a[i] = a[i - m];
        else
            a[i] = b[m - i - 1];
    }
    for (int i = 0; i < n; i++) {
        if (i != 0)
            cout << " ";
        cout << a[i];
    }
    return 0;
}

柳神

使用三次翻转达到循环位移的目的,末尾无空格通过最后一个数单独输出实现

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
#include <iostream>
#include <algorithm>
#include <vector>
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(begin(a), begin(a) + n);
        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;
}