原题传送门
给你一个数组,将数组中的元素向右轮转 k 个位置,其中 k 是非负数。
示例 1:
输入: nums = [1,2,3,4,5,6,7], k = 3
输出: [5,6,7,1,2,3,4]
解释:
向右轮转 1 步: [7,1,2,3,4,5,6]
向右轮转 2 步: [6,7,1,2,3,4,5]
向右轮转 3 步: [5,6,7,1,2,3,4]
示例 2:
输入:nums = [-1,-100,3,99], k = 2
输出:[3,99,-1,-100]
解释:
向右轮转 1 步: [99,-1,-100,3]
向右轮转 2 步: [3,99,-1,-100]
//超时
while(k--)
{
int ret = nums[numsSize - 1];
for(int i = numsSize - 1;i > 0; i--)
nums[i] = nums[i - 1];
nums[0] = ret;
}
首先第一个思路是使用空间换时间的思想,也就是将每次移动的数字放入另一个数组,最后拷贝回来
我们来看一下代码
void rotate(int* nums, int numsSize, int k){
int n = numsSize;
int tmp[n]; //变长数组【C99】
//1.首先对k做一个取模操作,防止数组访问越界
k %= n;
//2.将后k个数字移动到前面
int j = 0;
for(int i = n - k;i < n; ++i)
tmp[j++] = nums[i];
//3.n - k个数字移动到后面
for(int i = 0;i < n - k; ++i)
tmp[j++] = nums[i];
//4.将移动完的数组再拷贝绘原数组
for(int z = 0;z < n;++z)
nums[z] = tmp[z];
}
力扣提交结果
很明显,内存消耗比较多,因为需要维护一个临时数组
然后我们对这段代码来详细讲解一下
//1.首先对k做一个取模操作,防止数组访问越界
k %= n;
//2.将后k个数字移动到前面
int j = 0;
for(int i = n - k;i < n; ++i)
tmp[j++] = nums[i];
//3.n - k个数字移动到后面
for(int i = 0;i < n - k; ++i)
tmp[j++] = nums[i];
//4.将移动完的数组再拷贝绘原数组
for(int z = 0;z < n;++z)
nums[z] = tmp[z];
但是你会发现,就这样提交会出现执行出错,会出现一个溢出的情况,这是为什么呢?这个时候就需要使用到我开头跳过的那句代码
//1.首先对k做一个取模操作,防止数组访问越界
k %= n;
接下来再介绍一种比较巧妙的办法,就是【三步翻转法】
先看一下整体的代码
class Solution {
private:
void trans(vector<int>&nums, int begin, int end)
{
for(int i = begin, j = end; i <= j; ++i, --j)
swap(nums[i],nums[j]);
}
public:
void rotate(vector<int>& nums, int k) {
int n = nums.size();
k %= n;
//1.首先将前n-k个数字翻转
trans(nums, 0, n - k - 1);
//2.然后将后k个数字翻转
trans(nums, n - k, n - 1);
//3.最后将整体做一个翻转
trans(nums, 0, n - 1);
}
};
力扣提交结果
可以看出,整体的效率较新开辟一个临时数组来的高效一些
来分析一下代码
void trans(vector<int>&nums, int begin, int end)
{
for(int i = begin, j = end; i <= j; ++i, --j)
swap(nums[i],nums[j]);
}
以上就是本文的所有内容,如有问题请于评论区留言或者私信我,感谢您对本文的观看