EchoDemo's Blogs

PAT B1008 数组元素循环右移问题

一个数组A中存有N(N>0)个整数,在不允许使用另外数组的前提下,将每个整数循环向右移M(M>=0)个位置,即将A中的数据由(A0 A1……AN-1)变换为(AN-M …… AN-1 A0 A1……AN-M-1)(最后M个数循环移至最前面的M个位置)。如果需要考虑程序移动数据的次数尽量少,要如何设计移动的方法?

输入格式:每个输入包含一个测试用例,第1行输入N ( 1<=N<=100)、M(M>=0);第2行输入N个整数,
之间用空格分隔。

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

输入样例:
6 2
1 2 3 4 5 6

输出样例:
5 6 1 2 3 4

题目代码:本题需要注意的是当移动的次数为0,为N或者大于N的特殊情况。

#include<iostream>
#include<vector>
#include<algorithm>
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];
  }

  if (M != 0 && M != N) {
    if (M > N) M = M%N;
    reverse(a.begin(), a.begin() + N - M);//将前面的N-M位进行反转
    reverse(a.begin() + N - M, a.end());//将后面的M位进行反转
    reverse(a.begin(), a.end());//将整个数组进行反转
  }

  for (int i = 0;i < N;i++) {
    cout << a[i];
    if (i < N - 1) cout << " ";
  }
  return 0;
}
🐶 您的支持将鼓励我继续创作 🐶
-------------本文结束感谢您的阅读-------------