1008 数组元素循环右移问题
引入:
一个数组A中存有N(>0)个整数,在不允许使用另外数组的前提下,将每个整数循环向右移M(≥0)个位置,即将A中的数据由
变换为(
(最后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
!结尾无空行
语言:
c++
分析: 需要用reverse()函数对数列进行三次倒置,第一次将整列数倒置,第二次将从头到M-1位的数倒置(类似于区间的左开右闭,即不包括M),第三次将M位的数到数列尾部倒置。
例如:
输入6 2
1 2 3 4 5 6
第一次倒置:6 5 4 3 2 1
第二次倒置:5 6 4 3 2 1
第三次倒置:5 6 1 2 3 4
#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];
M = M % N;
if (M != 0) {
reverse(a.begin(), a.end());
reverse(a.begin(), a.begin() + M);
reverse(a.begin() + M, a.end());
}
for (int i = 0; i < N-1; i++) cout << a[i] << " ";
cout << a[N - 1];
return 0;
}