Given an array, rotate the array to the right by k
steps, where k
is non-negative.
Example 1:
Input: nums = [1,2,3,4,5,6,7], k = 3 Output: [5,6,7,1,2,3,4] Explanation: rotate 1 steps to the right: [7,1,2,3,4,5,6] rotate 2 steps to the right: [6,7,1,2,3,4,5] rotate 3 steps to the right: [5,6,7,1,2,3,4]
Example 2:
Input: nums = [-1,-100,3,99], k = 2 Output: [3,99,-1,-100] Explanation: rotate 1 steps to the right: [99,-1,-100,3] rotate 2 steps to the right: [3,99,-1,-100]
Constraints:
1 <= nums.length <= 105
-231 <= nums[i] <= 231 - 1
0 <= k <= 105
Follow up:
- Try to come up with as many solutions as you can. There are at least three different ways to solve this problem.
- Could you do it in-place with
O(1)
extra space?
因为题目中问我们能否考虑到使用o(1)的额外空间,本人首先考虑到的是使用暴力的方法,代码如下:
class Solution {
public:
void rotate(vector<int>& nums, int k) {
int temp;
int len = nums.size();
for(int i = 0;i < k;i ++)
{
temp = nums[len - 1];
for(int i = len - 1;i >= 1;i--)
{
nums[i] = nums[i-1];
}
nums[0] = temp;
}
}
};
时间复杂度为o(n*k),空间复杂度为o(1),n为nums数组长度,在数据规模过大时,发生超时。
下面先来考虑一般的使用额外空间的算法
class Solution {
public:
void rotate(vector<int>& nums, int k) {
int temp = 1;
int len = nums.size();
vector<int> ans;
if(k > len) temp = k / len + 1;
for(int i = 0;i < len;i ++)
{
ans.push_back(nums[(i + len * temp - k)%len]);
}
for(int i = 0;i < len;i ++)
{
nums[i] = ans[i];
}
}
};
这里使用的temp变量是为了防止当k大于数组长度len时数组元素赋值时出现越界情况(这主要是编译器的原因,如本人电脑使用的vs里面进行-1%3的结果是2,不需要使用到temp变量,而在leetcode中结果为-1)。
这里的赋值语句可以优化为官方题解中的语句:
ans[(i + k) % n] = nums[i];
考虑方向不同,效果一样,而且不需要使用temp变量防止下标出现负数。
同时最后的把ans中元素赋值到nums的循环可以使用vector自带的assign函数:
nums.assign(ans.begin(), ans.end());
代码更为简洁,assign函数的使用方法为:
函数原型:
void assign(const_iterator first,const_iterator last);
void assign(size_type n,const T& x = T());
功能:
将区间[first,last)的元素赋值到当前的vector容器中,或者赋n个值为x的元素到vector容器中,这个容器会清除掉vector容器中以前的内容。
比较有难度的是下面这种方法:(参考Demigodliu大佬在本题的题解)
思路如下:
- 首先对整个数组实行翻转,这样子原数组中需要翻转的子数组,就会跑到数组最前面。
- 这时候,从 k 处分隔数组,左右两数组,各自进行翻转即可。
class Solution {
public:
void rotate(vector<int>& nums, int k) {
reverse(nums.begin(),nums.end());
k = k % nums.size();
reverse(nums.begin(), nums.begin() + k);
reverse(nums.begin() + k,nums.end());
}
};
该算法时间复杂度为O(n),空间复杂度为O(1),题解里面对这个流程画了动画,比较详细,感兴趣的童鞋可以看看。
该算法不仅算法效率高,复现后的代码也很简洁,这里介绍一下使用到的reverse函数:
- 标准C中是没有reverse()函数,reverse函数使用时需要包含头文件#include <algorithm>
- reverse函数用于反转在[first,last)范围内的顺序(包括first指向的元素,不包括last指向的元素),注意这里的first和last都是指的地址,不是序号。
- reverse函数没有返回值。
有些函数使用时是放在元素前面,如这里的reverse函数,如sort函数,但有些函数是放在元素后面的,如begin函数,clear函数,end函数,以及上面提到的assign函数,这个我还没有弄得很懂,后面做个专题讨论一下。
但是这个方法的原理我没有弄得很懂,比如说为什么翻转后的数组和向右平移的数组保持一致,其背后有没有什么深层次的原理什么的,还有待去仔细讨论,同时如果有大佬能帮我解答这个困惑就更好了hh。