给定一个整数数组 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) 的 原地 算法解决这个问题吗?
题目如上:
解法如下
public void rotate(int[] nums, int k) {
// 获取输入数组的长度
int n = nums.length;
// 创建一个新的整型数组,用于存放旋转后的结果
int[] newArr = new int[n];
// 遍历原始数组
for (int i = 0; i < n; ++i) {
// 将原始数组中的元素按照旋转规则放入新数组中
// (i + k) % n 确保了索引不会超出范围,并实现了循环的效果
newArr[(i + k) % n] = nums[i];
}
// 使用 arraycopy 方法将新数组的内容复制回原数组
// 这里从新数组的第一个元素开始,复制 n 个元素到原数组中
System.arraycopy(newArr, 0, nums, 0, n);
}
很多人跟我一样,不太明白为什么
(i + k) % n 这个能举个例子为啥可以 确保了索引不会超出范围,并实现了循环的效果
原理如下
当然可以。假设我们有一个数组 nums,其长度为 n,并且我们要将它向右旋转 k 个位置。
示例说明
数组:nums = [1, 2, 3, 4, 5]
数组长度:n = 5
旋转步数:k = 2
计算过程
原始索引 i = 0:
新索引计算:(0 + 2) % 5 = 2
结果:nums[2] 应赋值为 nums[0] 的值,即 1
原始索引 i = 1:
新索引计算:(1 + 2) % 5 = 3
结果:nums[3] 应赋值为 nums[1] 的值,即 2
原始索引 i = 2:
新索引计算:(2 + 2) % 5 = 4
结果:nums[4] 应赋值为 nums[2] 的值,即 3
原始索引 i = 3:
新索引计算:(3 + 2) % 5 = 0
结果:nums[0] 应赋值为 nums[3] 的值,即 4
原始索引 i = 4:
新索引计算:(4 + 2) % 5 = 1
结果:nums[1] 应赋值为 nums[4] 的值,即 5
为什么 (i + k) % n 可以确保索引不会超出范围?
当 i + k 大于等于 n 时,取模运算会将其值限制在 [0, n-1] 范围内。
当 i + k 小于 n 时,取模运算的结果就是 i + k 本身,仍然在 [0, n-1] 范围内。
因此,无论 i + k 的值是多少,通过取模 % n 都能保证结果在有效的索引范围内。
旋转效果
最终得到的数组为 [4, 5, 1, 2, 3],可以看到数组中的每个元素都向右移动了两个位置,实现了旋转的效果。
这就是 (i + k) % n 如何确保索引不会超出范围,并实现循环效果的过程。