LeetCode_189.轮转数组

目录

一、题目

二、解法

1）BF法

2）新数组

3) 逆置法

---

一、题目

给你一个数组，将数组中的元素向右轮转 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);
}