Rotate an array of n elements to the right by k steps.
For example, with n = 7 and k = 3, the array
[1,2,3,4,5,6,7]
is rotated to [5,6,7,1,2,3,4]
.
Note:
Try to come up as many solutions as you can, there are at least 3 different ways to solve this problem.
Code (Java):Try to come up as many solutions as you can, there are at least 3 different ways to solve this problem.
public class Solution { public void rotate(int[] nums, int k) { if (nums == null || nums.length <= 1 || k <= 0) { return; } // Step 1: swap each element of the array int i = 0; int j = nums.length - 1; while (i < j) { swap(nums, i, j); i++; j--; } k %= nums.length; // Step 2: swap the first k elements i = 0; j = k - 1; while (i < j) { swap(nums, i, j); i++; j--; } // Step 3: swap the rest of k elements i = k; j = nums.length - 1; while (i < j) { swap(nums, i, j); i++; j--; } } private void swap(int[] nums, int i, int j) { int tmp = nums[i]; nums[i] = nums[j]; nums[j] = tmp; } }
No comments:
Post a Comment