经典题型---旋转数组

经典题型—旋转数组

一、题目

给定一个整数数组 nums,将数组中的元素向右轮转 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]

提示:

1 <= nums.length <= 105

-231 <= nums[i] <= 231 - 1

0 <= k <= 105

题目理解

右旋可以理解成数组里的元素向右挪动k个单位,而出界的元素再按照正序补到前面的空位。

在这里插入图片描述
但是如果当k > numsSize的时候,那就需要使用到余数了,k % numsSize,直到k小于numsSize就可以进行程序操作了。

二、代码实现

法一:数组翻转法

void rotate(int* nums, int numsSize, int k) {
    int i = 0, j = 0;
    for (i = 0, j = numsSize - k - 1; i <= j; i++, j--)
    {
        int tmp = nums[i];
        nums[i] = nums[j];
        nums[j] = tmp;
    }
    for (i = numsSize - k, j = numsSize - 1; i <= j; i++, j--)
    {
        int tmp = nums[i];
        nums[i] = nums[j];
        nums[j] = tmp;
    }
    for (i = 0; i < numsSize; i++)
    {
        int tmp = nums[i];
        nums[i] = nums[j];
        nums[j] = tmp;
    }
}

优化

void Reverse(int* nums, int begin, int end)
{
    int i = 0, j = 0;
    for (i = begin, j = end; i <= j; i++, j--)
    {
        int tmp = nums[i];
        nums[i] = nums[j];
        nums[j] = tmp;
    }
}
void rotate(int* nums, int numsSize, int k)
{
    int i = 0, j = 0;
    Reverse(nums, 0, numsSize - k - 1);
    Reverse(nums, numsSize - k, numsSize - 1);
    Reverse(nums, 0, numsSize - 1);
}

不难看出时间复杂度是O(2n),即O(n)

法二:临时数组法

void rotate(int* nums, int numsSize, int k) 
{
    int arr[100000] = { 0 };
    if (0 == k)
    {
        ;
    }
    else
    {
        k %= numsSize;
        while (k >= numsSize)
        {
            k %= numsSize;
        }
        int i = 0, j = 0;
        for (i = 0, j = numsSize - k; i < k, j <= numsSize - 1; i++, j++)
        {
            arr[i] = nums[j];
        }
        for (i = k, j = 0; i <= numsSize - 1, j < numsSize - k; i++, j++)
        {
            arr[i] = nums[j];
        }
        for (i = 0, j = 0; i < numsSize, j < numsSize; i++, j++)
        {
            nums[i] = arr[j];
        }
    }
}

优化

void rotate(int* nums, int numsSize, int k) {
    int newArr[numsSize];//变长数组不能初始化
    for (int i = 0; i < numsSize; ++i) {
        newArr[(i + k) % numsSize] = nums[i];
    }
    for (int i = 0; i < numsSize; ++i) {
        nums[i] = newArr[i];
    }
}

这个方法在使用时,笔者掉入了一个很危险的错误,和大家分享出来,希望大家在以后的编程上避开这个坑。**

错误示范:

void rotate(int* nums, int numsSize, int k) {//返回栈空间地址的问题
    int arr1[100000] = { 0 };
    if (0 == k)
    {
        ;
    }
    else
    {
        k %= numsSize;
        while (k >= numsSize)
        {
            k %= numsSize;
        }
        int i = 0, j = 0;
        for (i = 0, j = numsSize - k; i < k, j <= numsSize - 1; i++, j++)
        {
            arr1[i] = nums[j];
        }
        for (i = k, j = 0; i <= numsSize - 1, j < numsSize - k; i++, j++)
        {
            arr1[i] = nums[j];
        }
    }
    nums = arr1;
}

这是一个非常典型的返回栈空间(临时变量地址)的错误,在执行函数过程中,创建了一个临时数组,这个数组存储的就是满足输出形式的数组,但是当函数调用结束后,这块空间就被销毁了,nums反而成了野指针,所以这样的问题一定要避免。

在返回地址的时候要十分小心

评论 9
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值