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