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

题目:

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& nums, int k) {
        int n=nums.size();
        int l=k%n;
        int* temp=new int[n];
        if(l>0){
          for(int i=0;i
后来想了一下是不是可以memcpy()函数来实现上述过程,于是写了下面的代码:

class Solution {
public:
    void rotate(vector& 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;
    }
};
但是提交的时候发现出现了下述错误:

【LeetCode-189】Rotate Array(C++)_第1张图片

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

void rotate(vector& nums, int k) 

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

class Solution {
public:
    void rotate(vector& 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& arr,int start,int end){
        int i=start;
        int j=end;
        int temp;
        for(i,j;i& 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;
    }
};




你可能感兴趣的:(Leetcode)