轮转数组(Java)
给你一个数组,将数组中的元素向右轮转 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]
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/rotate-array
今天在LeetCode上写轮转数组这一题,我自己写的在idea上也能运行,输出,没毛病,死活在LeetCode上不通过,后来在网上也看了一些代码,找到了一种比较高效的在最下面:
下面是我自己写的代码(LeetCode上没通过那个)
public class Main54 { public static void main(String[] args) { int[] nums = {1, 2, 3, 4, 5, 6, 7, 8,9,10}; int k = 3; int[] x = new int[nums.length]; int l = nums.length-k; for(int i = 0 ;i < k;i++){ x[i] = nums[l]; l++; } int n = 0; for (int d = k;d < nums.length; d++ ){ x[d] = nums[n]; n++; } for(int q = 0;q < x.length;q++){ System.out.print(x[q] + " "); } } }
回来实在没办法了,在CSDN上看了一些代码,
发现MOD比较好用(利用余数轮转数组,顶级思路)
(这个是LeetCode上通过的)
public void rotate(int[] nums, int k) { int len = nums.length; int[] arr = new int[len]; for(int i = 0;i < len;i++){ // (i + k)%len利用余数轮转数组,顶级思路 arr[(i + k)%len] = nums[i]; } //把arr数组赋值给nums数组 for(int i = 0;i < len;i++){ nums[i] = arr[i]; } }