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

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

一个数组A中存有N（>）个整数，在不允许使用另外数组的前提下，将每个整数循环向右移M（≥）个位置，即将A中的数据由（A0A1⋯AN−1）变换为（AN−M⋯AN−1A0A1⋯AN−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 = ; i < N; ++i)

     {

         cin >> v[i];

     }

     M %= N;    //防止M比N大

     if (M != )

     {

         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() - ; ++it)

     {

         cout << *it << " ";

     }

     cout << *it;

     return ;

 }

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]);

             }

         }

     }

 }

巴特西