PAT 乙级 1008.数组元素循环右移问题 C++/Java

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

 题目来源

一个数组A中存有N（>）个整数，在不允许使用另外数组的前提下，将每个整数循环向右移M（≥）个位置，即将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）和M（≥）；第2行输入N个整数，之间用空格分隔。

输出格式:

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

输入样例:

6 2
1 2 3 4 5 6

输出样例:

5 6 1 2 3 4

C++实现：

分析:

实现数组右移分3步：

（1）将数组颠倒（2）将数组 [ 0, m ) 颠倒（3）将数组 [ m, length) 颠倒

注意：

（1）右移位数 M 可能会比元素个数 N 大,要先进行 M % N 的操作

（2）判断 M 是否为 0 

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

int main()
{
    //先将数组倒置，然后再将数组后的M位再倒置
    int N, M;
    cin >> N >> M;    //元素个数，后移位数
    vector<int> v(N);
    for (int i = 0; i < N; ++i)
    {
        cin >> v[i];
    }

    M %= N;    //防止M比N大
    if (M != 0)
    {
        reverse(v.begin(), v.begin() + N);    //第一次将所有数组倒置
                                            //由1-2-3-4-5-6变成6-5-4-3-2-1

        reverse(v.begin(), v.begin() + M);    //第二次将部分数组倒置
                                            //由6-5-4-3-2-1变成5-6-4-3-2-1

        reverse(v.begin() + M, v.end());    //第三次将部分数组倒置
                                            //由6-5-4-3-2-1变成5-6-1-2-3-4
    }
    vector<int>::iterator it;

    for (it = v.begin(); it != v.end() - 1; ++it)
    {
        cout << *it << " ";

    }
    cout << *it;
    return 0;
}

 

Java实现：

 

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner input = new Scanner(System.in);
        int n = input.nextInt();
        int m = input.nextInt();
        int a[] = new int[n];
        int i = 0;
        for (i = 0; i < n; i++) {
            a[i] = input.nextInt();
        }
        if (n < m) {
            m = m % n;
        }
        if (n == m) { 
            for (i = 0; i < n; i++) {
                if (i != 0) {
                    System.out.print(" ");
                }
                System.out.print(a[i]);
            }
        } else {
            for (int j = n - m; j < n; j++) {
                System.out.print(a[j] + " ");
            }
            for (int j = 0; j < n - m; j++) {
                if (j != 0) {
                    System.out.print(" ");
                }
                System.out.print(a[j]);
            }
        }
    }
}