前言
今天是坚持训练算法题的第三天,感觉越来越有意思了,今天做的这道旋转数组貌似是408考研的真题,因为考研复习的时候就已经很对于这种题目很熟悉了,所以今天做起来很顺利。如果不出意外的话,晚上我还会在做一道然后继续发博客。
题目
大家想看题目和思路的话详情可以参考小浩算法,讲的很好。
小浩算法 – 旋转数组
思路及代码
我的思路
这个思路其实只要见过一次就不会忘了,这种循环的效果其实只需要三步
就可以实现:
- 逆置
前半部分
数组 - 逆置
后半部分
数组 - 逆置
整个
数组
我们发现最核心的就是写这个数组逆置的函数,我觉得函数的核心也有以下几点:
- 形参怎么选?
由于要求空间复杂度为O(1)
,我们其实不需要再考虑数组的划分了,就算只逆置部分数组也可以把整个数组放进去
。这里还可以采用引用
,那么就不用返回值,内存也不用再为vector开辟空间了(值得注意的是,只有在形参里需要写&引用标识,在调用函数的时候是不需要的),最后还需要起始位置和结束位置。
void reverseArray(vector<int> &arr, int begin, int end)
- 怎么在空间复杂度为1的条件下实现逆置?
只需要数组的前半部分和后半部分对应的元素交换
就好。不管是数组长度是奇数还是偶数,都需要从第一个遍历到(总数/2)的位置。大家可以看着我的参数好好想一想。
//这里如果i其实最后取到了begin + ((end - begin + 1) / 2) -1 ,因为是小于号
i < begin + ((end - begin + 1) / 2)
我的代码
#include<iostream>
#include<vector>
using namespace std;
void reverseArray(vector<int> &arr, int begin, int end) {
int wrap;
for (int i = begin; i < begin + ((end - begin + 1) / 2); i++) {
wrap = arr[i];
arr[i] = arr[end - (i - begin)];
arr[end - (i - begin)] = wrap;
}
}
int main() {
vector<int> intVec;
int n;
cout << "请输入数组中元素的个数" << endl;
cin >> n;
cout << "请依次输入每个元素" << endl;
int x;
for (int i = 0; i < n; i++) {
cin >> x;
intVec.push_back(x);
}
cout << "请输入旋转数" << endl;
int reverseBit;
cin >> reverseBit;
reverseArray(intVec, 0, intVec.size() - reverseBit - 1);
reverseArray(intVec, intVec.size() - reverseBit, intVec.size() - 1);
reverseArray(intVec, 0 , intVec.size()-1 );
for (int i = 0; i < intVec.size(); i++) {
cout << intVec[i] << " ";
}
cout << endl;
}