【LeetCode-189】Rotate Array(C++)

来源:互联网 发布:数控编程坐标系 编辑:程序博客网 时间:2024/06/14 01:13

题目:

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.我的想法是创建一个新数组,然后将原数组中的后k个元素放到新数组的前k个位置,将原数组中的前n-k个元素放到新数组的后n-k个位置,然后将新数组的元素复制到旧数组中去,我的代码:

class Solution {public:    void rotate(vector<int>& nums, int k) {        int n=nums.size();        int l=k%n;        int* temp=new int[n];        if(l>0){          for(int i=0;i<n-l;i++){             temp[i+l]=nums[i];        }        for(int i=n-l;i<n;i++){            temp[i-n+l]=nums[i];        }        for(int i=0;i<n;i++){            nums[i]=temp[i];        }        }        delete[] temp;    }};
后来想了一下是不是可以memcpy()函数来实现上述过程,于是写了下面的代码:

class Solution {public:    void rotate(vector<int>& nums, int k) {        int n=nums.size();        k=k%n;        int* temp=new int[n];        memcpy(temp,nums+(n-k),sizeof(int)*k);        memcpy(temp+k,nums,sizeof(int)*(n-k));        memcpy(nums,temp,sizeof(int)*n);        delete[] temp;    }};
但是提交的时候发现出现了下述错误:


查了一下资料发现,由于函数列表里面是这么定义nums的:

void rotate(vector<int>& nums, int k) 

nums是一个vector容器的名称,而不是一个数组,所以nums并不代表数组的第一个元素的地址,但由于vector里面的元素是连续存储的,就相当于一个动态数组,故将代码改成下面这种形式,就通过了:

class Solution {public:    void rotate(vector<int>& nums, int k) {        int n=nums.size();        k=k%n;        int* temp=new int[n];        memcpy(temp,&nums[0]+(n-k),sizeof(int)*k);        memcpy(temp+k,&nums[0],sizeof(int)*(n-k));        memcpy(&nums[0],temp,sizeof(int)*n);        delete[] temp;    }};
2.第二种方法是一种很神奇的方法,需要三次逆转操作:假设一个数组中有7个元素,{1,2,3,4,5,6,7},k=3。第一步,将整个数组逆转,得到{7,6,5,4,3,2,1};第二步,将数组中的前k个元素逆转,得到{5,6,7,4,3,2,1};第三步,将数组中的后n-k个元素逆转,得到{5,6,7,1,2,3,4}。代码如下:

class Solution {public:    void reverse(vector<int>& arr,int start,int end){        int i=start;        int j=end;        int temp;        for(i,j;i<j;i++,j--){            temp=arr[j];            arr[j]=arr[i];            arr[i]=temp;        }        return;    }    void rotate(vector<int>& nums, int k) {        int n=nums.size();        k=k%n;        reverse(nums,0,n-1);        reverse(nums,0,k-1);        reverse(nums,k,n-1);        return;    }};




0 0