leetcode刷题记录C语言-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

代码

递归：

int move(int* nums, int numsSize, int origin, int k, int n, int target){
    if(target == origin){
        nums[origin] = n;
        return 1;
    }
    int tmp = nums[target];
    nums[target] = n;
    n = tmp;
    return move(nums, numsSize, origin, k, n, (target+k)%numsSize)+1;
}
void rotate(int* nums, int numsSize, int k) {
    int i = 0;
    int cnt = 0;
    int n = 0;
    while(cnt != numsSize){
        n = nums[i];
        cnt += move(nums, numsSize, i, k, n, (k+i)%numsSize);
        i++;
    }
}

迭代：

void rotate(int* nums, int numsSize, int k) {
    int idex = 0;
    int cnt = 0;
    int n = 0;
    // 移动次数判断
    while(cnt != numsSize){
    	// idex为移动起点
        n = nums[idex];
        int target = (idex+k)%numsSize;
        // 移动回到起点
        while(target != idex){
            int tmp = nums[target];
            // 移动
            nums[target] = n;
            n = tmp;
            target = (target+k)%numsSize;
            cnt++;
        }
        nums[target] = n;
        cnt++;
        idex++;
    }
}

代码思路：记录移动前的数值，移动，找到下一次移动的位置。

注意：一次移动可能不会完成整个数组移动，需要修改起点继续移动完成。

总结

扩展：修改代码中的加减，右移可以改成左移。