LeetCode OJ: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].

将数组的内容倒置,看例子就知道是什么意思:

 1 class Solution {
 2 public:
 3     void rotate(vector<int>& nums, int k) {
 4         if(k > nums.size()) k %= nums.size();//这里要注意,k的大小可能比size要大一点
 5         vector<int> tmpVec1{nums.begin(), nums.begin() + (nums.size() - k)};
 6         vector<int> tmpVec2{nums.begin() + (nums.size() - k), nums.end()};
 7         int sz1 = tmpVec1.size();
 8         for(int i = 0; i < sz1; ++i){
 9             tmpVec2.push_back(tmpVec1[i]);
10         }
11         nums = tmpVec2;
12     }
13 };

上面这个是数组拷贝的方法,不过速度比较慢一点,一开始用一个二重循环来解决,不过那样总是超时,所以又想了上面这个方法。

还有题目说了希望可以用到O(1)的额外空间。看了下别人写的:炒鸡简单啊

 1 class Solution {
 2 public:
 3     void rotate(vector<int>& nums, int k) {
 4         int sz = nums.size();
 5         k %= sz;
 6         reverse(nums.begin(), nums.begin() + (sz - k));
 7         reverse(nums.begin() + (sz - k), nums.end());
 8         reverse(nums.begin(), nums.end());
 9     }
10 };

先把前面的部分反转,再将后面的部分反转,最后再做一次反转就可以了。

 java接比较坑爹啊,还要自己手写反转函数,额,这一点没有STL好用了,可能只是我不知道罢了,代码如下:

 1 public class Solution {
 2     public void rotate(int[] nums, int k) {
 3         k = k % nums.length;
 4         rev(nums, nums.length-k, nums.length - 1);
 5         rev(nums, 0, nums.length-k-1);
 6         rev(nums, 0, nums.length - 1);
 7     }
 8     
 9     //自己要单独的写一个rev函数
10     public void rev(int [] arr, int start, int end){
11         int tmp = 0;
12         while(start < end){
13             tmp = arr[start];
14             arr[start] = arr[end];
15             arr[end] = tmp;
16             start++;
17             end--;
18         }
19     }
20 }

 

posted @ 2015-10-19 19:23  eversliver  阅读(396)  评论(0编辑  收藏  举报