189. 轮转数组

189. 轮转数组

中等

相关标签

相关企业

提示

给定一个整数数组 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

进阶：

• 尽可能想出更多的解决方案，至少有 三种 不同的方法可以解决这个问题。
• 你可以使用空间复杂度为 O(1) 的 原地 算法解决这个问题吗？

题解：

看答案采用三次reverse的方法，这样空间复杂度为O(1)。reverse的范围和函数定义方式很特别。很好用。

#include <stdio.h>
#include <stdlib.h>

void reverse (int* nums, int start, int end)
{
    int i, j, temp;
    i = start;
    j = end;
    while (i < j)
    {
        temp = nums[i];
        nums[i] = nums[j];
        nums[j] = temp;
        i ++;
        j --;
    }
    return;
}

int main()
{
    int size, k, i;
    scanf("%d", &size);
    int* nums = (int*)malloc(sizeof(int) * size);
    for (i = 0;i <= size - 1;i ++)
    {
        scanf("%d", &nums[i]);
    }
    scanf("%d", &k);
    reverse(nums, 0, size - 1 - k);
    reverse(nums, size - k, size - 1);
    reverse(nums, 0, size - 1);
    for (i = 0;i <= size - 1;i ++)
    {
        printf("%d ", nums[i]);
    }

    return 0;
}