该题我们之前也是做过的,
我们可以使用额外的数组来将每个元素放至正确的位置。用 nn 表示数组的长度,我们遍历原数组,将原数组下标为 ii 的元素放至新数组下标为 (i+k)\bmod n(i+k)modn 的位置,最后将新数组拷贝至原数组即可
如下:
class Solution {
public:
void rotate(vector<int>& nums, int k) {
vector<int> result(nums.size(),0);
int n=nums.size();
for(int i=0;i<n;i++)
{
result[(i+k)%n]=nums[i];
}
for(int i=0;i<n;i++)
{
nums[i]=result[i];
}
}
};
主要我们要介绍下面这种新方法——数组翻转
该方法基于如下的事实:当我们将数组的元素向右移动 k 次后,尾部 k mod n 个元素会移动至数组头部,其余元素向后移动 k mod n 个位置。
该方法为数组的翻转:我们可以先将所有元素翻转,这样尾部的k mod n 个元素就被移至数组头部,然后我们再翻转 [0, k\bmod n-1]区间的元素和 [k\bmod n, n-1]区间的元素即能得到最后的答案。
我们以 n=7,k=3 为例进行如下展示:
class Solution {
public:
void reverse(vector<int>& nums, int start, int end) {
while (start < end) {
swap(nums[start], nums[end]);
start += 1;
end -= 1;
}
}
void rotate(vector<int>& nums, int k) {
k %= nums.size();
reverse(nums, 0, nums.size() - 1);
reverse(nums, 0, k - 1);
reverse(nums, k, nums.size() - 1);
}
};