1008 数组元素循环右移问题 (20)
时间限制
400 ms
内存限制
65536 kB
代码长度限制
8000 B
判题程序
Standard
一个数组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
经人启发,终于想起来了C++还有insert()这么个神奇的东西,之前用交换的方法的纠结一扫而光,直接实现。
#include <iostream>
#include <vector>
using namespace std;
int main (void){
int N = 0, M = 0, num = 0;
cin >> N >> M;
vector<int> ivec;
for (int count = 0; count < N; count++){
cin >> num;
ivec.push_back(num);
}
for (int count = 0; count < M; count++){
ivec.insert(ivec.begin(), *(ivec.end()-1));
ivec.pop_back();
}
for (vector<int>::iterator iter = ivec.begin(); iter < ivec.end(); iter++){
cout << *iter;
if (iter+1 != ivec.end())
cout << " ";
}
return 0;
}
虽然在这里看起来时间复杂度好像是O(n),但是仔细一想就会发现,每次insert之后,后面所有的元素都要后移,因此实际上这个算法的复杂度应该是O(n²)(这样一想还是有点沮丧)。
另外,在这里
for (int count = 0; count < M; count++){
ivec.insert(ivec.begin(), *(ivec.end()-1));
ivec.pop_back();
}
如果把条件改成count < N-M,输出就会变成向左移动M位(别问我是怎么知道的。。。。)