【刷题】leetcode 189. 轮转数组

leetcode 189. 轮转数组

  • 一、题目描述
  • 二、示例
  • 三、实现
    • 方法1 暴力求解
    • 方法2 空间换时间
    • 方法3 三次逆置


leetcode 189. 轮转数组

一、题目描述

给定一个整数数组 nums,将数组中的元素向右轮转 k 个位置,其中 k 是非负数。

二、示例

输入: 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]

三、实现

方法1 暴力求解

数组每次右旋一个数,右旋k次

1.保存最后一个数
2.前面所有数字右移
3.将保存的最后一个数放到第一个位置

void rotate(int* nums, int numsSize, int k) {
	k %= numsSize;
	for (int i = 0; i < k; ++i) {
		// 1.保存最后一个数
		int temp = nums[numsSize - 1];
		// 2.前面所有数字右移
		for (int j = numsSize - 2; j >= 0; j--) {
			nums[j + 1] = nums[j];
		}
		// 3.将保存的最后一个数放到第一个位置
		nums[0] = temp;
	}
}

方法2 空间换时间

新申请一块空间,用于数据交换。

1.申请新空间
2.原数组前 size-k 个数移到新空间的后面
3.原数组最后 k 个数移到新空间的前面
4.新空间数据覆盖原数组

void rotate(int* nums, int numsSize, int k) {
	k %= numsSize;
	// 1.申请新空间
	int* tmp = (int*)malloc(numsSize * sizeof(int));
	// 2.原数组前 size-k 个数移到新空间的后面
	memcpy(tmp + k, nums, (numsSize - k) * sizeof(int));
	// 3.原数组最后 k 个数移到新空间的前面
	memcpy(tmp, nums + numsSize - k, k * sizeof(int));
	// 4.新空间数据覆盖原数组
	memcpy(nums, tmp, numsSize * sizeof(int));
	free(tmp);
	tmp = NULL;
}

方法3 三次逆置

原数组的下列三个区间数据逆置:
[0, size - k - 1]
[size - k, size - 1]
[0, size - 1]

// [begin, end]区间内数据逆置
void reverse(int* nums, int begin, int end) {
    while(begin < end) {
        int temp = nums[begin];
        nums[begin] = nums[end];
        nums[end] = temp;
        begin++;
        end--;
    }
}
void rotate(int* nums, int numsSize, int k){
    k %= numsSize;
   	// 1.[0, size - k - 1]逆置
    reverse(nums, 0, numsSize-k-1);
    // 2.[size - k, size - 1]逆置
    reverse(nums, numsSize-k, numsSize-1);
    // 3.整体逆置
    reverse(nums,0, numsSize-1);
}

你可能感兴趣的:(刷题,leetcode,算法,数据结构)