首页 > 图灵资讯 > 技术篇>正文
轮转数组
2023-05-15 09:36:35
给定一个整数数组 nums,向右旋转数组中的元素 k位置,其中k是非负数。
示例 1:
输入: nums = [1,2,3,4,5,6,7], k = 3
输出: [5,6,7,1,2,3
解释:
向右轮转 1 步: [7,1,2,3,4,5]
向右轮转 2 步: [6,7,1,2,4,5]
向右轮转 3 步: [5,6,7,1,2,3
示例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)的空间复杂度 原地算法解决这个问题吗?
上代码:
class Solution { private void reverse(int[] nums, int start, int end) { for (int i = start, j = end; i < j; i++, j--) { int temp = nums[j]; nums[j] = nums[i]; nums[i] = temp; } } public void rotate(int[] nums, int k) { int n = nums.length; k %= n; reverse(nums, 0, n - 1); reverse(nums, 0, k - 1); reverse(nums, k, n - 1); }}