目录
一、题目
给你一个数组,将数组中的元素向右轮转 k
个位置,其中 k
是非负数。
题目链接:189. 轮转数组 - 力扣(LeetCode) (leetcode-cn.com)
二、解法
1)BF法
即最暴力的解法:时间复杂度O(N^2) 空间复杂度O(1)
// BF法 时间复杂度O(N^2) 空间复杂度O(1)
void rotate01(int* nums, int numsSize, int k)
{
while (k--)
{
int tmp = nums[numsSize - 1];
int end = numsSize - 1;
while (end--)
{
nums[end + 1] = nums[end];
}
nums[0] = tmp;
}
}
2)新数组
即创建一个新数组,将旋转后的数字放入新数组里,在将新数组的内容拷贝给原数组:时间复杂度O(N) 空间复杂度O(N)
// 数组法 时间复杂度O(N) 空间复杂度O(N)
void rotate02(int* nums, int numsSize, int k)
{
k %= numsSize;
int* tmp = (int*)malloc(sizeof(int)*numsSize);
if (tmp == NULL)
{
exit(-1);
}
for (int i = 0; i < k; i++)
{
tmp[i] = nums[numsSize - k + i];
}
for (int j = 0; j < numsSize - k; j++)
{
tmp[k + j] = nums[j];
}
for (int i = 0; i < numsSize; i++)
{
nums[i] = tmp[i];
}
}
3) 逆置法
目前最优的解法:时间复杂度O(N) 空间复杂度 O(1)
将需要轮转的元素和不需要旋转的元素各自逆置一次,再将整个数组逆置,得到的就是目标数轮转后的数组
// 三次逆置法 时间复杂度O(N) 空间复杂度 O(1)
void reverse(int* a, int left, int right)
{
while (left < right)
{
int tmp = a[left];
a[left] = a[right];
a[right] = tmp;
left++;
right--;
}
}
void rotate03(int* nums, int numsSize, int k) {
k %= numsSize;
reverse(nums, numsSize - k, numsSize - 1);
reverse(nums, 0, numsSize - k - 1);
reverse(nums, 0, numsSize - 1);
}