问题描述:
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].
分析:这道题有个模糊的地方就是当k>numsSize时怎么处理,正确方法是k=k%numsSize,别问我为什么,我也不知道。
起初的代码如下:每次移除一个,然后再开始位置处添加一个。时间复杂度为O(kn),空间复杂度为O(1).结果超时。
void rotate(int* nums, int numsSize, int k) {
k %= numsSize;
for(int i = 0;i<k;i++){
int tmp = nums[numsSize-1];
for(int j = numsSize-1;j>0;j--){
nums[j] = nums[j-1];
}
nums[0] = tmp;
}
}
后来又改写为如下形式,即开辟了一块临时内存空间,时间复杂度为O(n),空间复杂度为O(k).
代码如下:8ms
void rotate(int* nums, int numsSize, int k) {
int *tmpNums = (int *)malloc(sizeof(int)*k);
k = k%numsSize;
memcpy(tmpNums, nums+numsSize-k, sizeof(int)*k);
memmove(nums + k, nums, sizeof(int)*(numsSize - k));
memcpy(nums, tmpNums, sizeof(int)*k);
}