题目:
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]
.
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;
}
};
但是提交的时候发现出现了下述错误:
查了一下资料发现,由于函数列表里面是这么定义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;
}
};