Leetcode: Rotate Array

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.



[show hint]



Hint:

Could you do it in-place with O(1) extra space?

Naive想法就是保存一个原数组的拷贝,然后把原数组分成前len-k个元素和后k个元素两部分,把后k个元素放到前len-k个元素前面去。这样做需要O(N)空间

in-place做法是: 

(1) reverse the array;

(2) reverse the first k elements;

(3) reverse the last n-k elements.

The first step moves the first n-k elements to the end, and moves the last k elements to the front. The next two steps put elements in the right order.

 1 public class Solution {

 2     public void rotate(int[] nums, int k) {

 3         int len = nums.length;

 4         k %= len;

 5         reverse(nums, 0, len-1);

 6         reverse(nums, 0, k-1);

 7         reverse(nums, k, len-1);

 8     }

 9     

10     public void reverse(int[] nums, int l, int r) {

11         while (l <= r) {

12             int temp = nums[l];

13             nums[l] = nums[r];

14             nums[r] = temp;

15             l++;

16             r--;

17         }

18     }

19 }

需要注意的是第4行,右移偏移量k可能比数组长度len要大,所以要先 k%=len;

你可能感兴趣的:(LeetCode)