1 题目描述(题目有坑,要注意)
一个数组A中存有 n 个整数,在不允许使用另外数组的前提下,将每个整数循环向右移 M( M >=0)个位置,即将A中的数据由(A0 A1 ……AN-1 )变换为(AN-M …… AN-1 A0 A1 ……AN-M-1 )(最后 M 个数循环移至最前面的 M 个位置)。如果需要考虑程序移动数据的次数尽量少,要如何设计移动的方法?
数据范围:0<n≤100,0≤m≤1000
进阶:空间复杂度 O(1),时间复杂度O(n)
示例1
输入:6,2,[1,2,3,4,5,6]
返回值:[5,6,1,2,3,4]
示例2
输入:4,0,[1,2,3,4]
返回值:[1,2,3,4]
2 思路
1.整体逆置;[0,n)
2.前m个逆置;[0,m)
3.m之后的元素逆置;[m,n)
3 实现
3.1 自己实现逆置函数
int reverse(int i,int j, vector<int>& arr){
for(; i < j; i++)
swap(arr[j--],arr[i]);
return 0;
}
vector<int> solve(int n, int m, vector<int>& a) {
reverse(0, n - 1, a);//1.整体逆置;[0,n)
reverse(0, m - 1, a);//2.前m个逆置;[0,m)
reverse(m, n - 1, a);//m之后的元素逆置;[m,n)
return a;
}
2个测试用例,全通过,但提交ton过滤2/17;
反复斟酌,思路没问题,逆置函数优化后,也没问题;
问题依旧,为何?
复杂度分析:
时间复杂度:O(n)O(n)O(n),三次reverse函数的复杂度都最坏为O(n)O(n)O(n)
空间复杂度:O(1)O(1)O(1),没有使用额外的辅助空间
3.2 使用C++ 库函数reverse() 头文件
reverse函数用于反转 [ first , last )范围内的顺序。(注意:左闭右开)
reverse()函数没有返回值。
标准C中没有此函数,C++新增,需要使用头文件
#include <algorithm>
可用于容器、数组、字符串等。
vector<int> solve(int n, int m, vector<int>& a) {
reverse(a.begin(),a.end());//1.整体逆置;[0,n)
reverse(a.begin(),a.begin() + m);//2.前m个逆置;[0,m)
reverse(a.begin() + m,a.begin() + n);//m之后的元素逆置;[m,n)
return a;
}
2个测试用例,全通过,但提交ton过滤2/17;
问题依旧,为何?
复杂度分析:
时间复杂度:O(n)O(n)O(n),三次reverse函数的复杂度都最坏为O(n)O(n)O(n)
空间复杂度:O(1)O(1)O(1),没有使用额外的辅助空间
3.3 终于发现了,坑!
数据范围:0<n≤100,0≤m≤1000,,,,
m可能大于n,
//m取余,因为每次长度为n的旋转数组相当于没有变化,最终变化的还是余数的那个m
m = m % n;
调整后代码
vector<int> solve(int n, int m, vector<int>& a) {
m %= n;//m可能大于n:0<n≤100,0≤m≤1000
reverse(a.begin(),a.end());
reverse(a.begin(),a.begin() + m);
reverse(a.begin() + m,a.begin() + n);
return a;
}