题目描述:给定一个整数数组 nums
,将数组中的元素向右轮转 k
个位置,其中 k
是非负数。
输入: nums = [1,2,3,4,5,6,7], k = 3 输出:[5,6,7,1,2,3,4]
向右轮转 1 步:[7,1,2,3,4,5,6]
向右轮转 2 步:[6,7,1,2,3,4,5]
向右轮转 3 步:[5,6,7,1,2,3,4]
一,方法一
时间复杂度为O(N^2), 空间复杂度为O(1)。
【1,2,3,4,5,6,7】---> 【7,1,2,3,4,5,6】
使用一个tmp变量将存储7,然后所有其他数字向右移动一次,再将7放在数组首位置。
二,方法二
时间复杂度O(N), 空间复杂度O(N)
【1,2,3,4, 5,6,7】----> 【5,6,7, 1,2,3,4】
开辟一块空间,把后k个数字之间拷贝到前面,把前面的n-k个拷贝到后面
void rotate(int* nums, int numsSize, int k){
int n = numsSize;
int* tmp = (int*) malloc(sizeof(int)*n);
k %= n;
memcpy(tmp, nums+n-k, sizeof(int)*k);
memcpy(tmp+k, nums, sizeof(int)*(n-k));
memcpy(nums, tmp, sizeof(int)*n);
free(tmp);
tmp = NULL;
}
三,方法三
时间复杂度O(N), 空间复杂度O(1)。
首先将前n-k个元素逆置,然后k个元素逆置,最后全部元素逆置。
void reverse(int* left,int* right)
{
while(left <= right)
{
int tmp = *left;
*left = *right;
*right = tmp;
left++;
right--;
}
}
void rotate(int* nums, int numsSize, int k){
int n =numsSize;
k %= n;
reverse(nums, nums+n-k-1);
reverse(nums+n-k,nums+n-1);
reverse(nums,nums+n-1);
}