方法1 (三次逆置法):
cpp
void reverse(int* nums, int start, int end) {
while (start < end) {
int temp = nums[start];
nums[start] = nums[end];
nums[end] = temp;
start++;
end--;
}
}
void rotate(int* nums, int numsSize, int k) {
k = k % numsSize; // 防止 k 大于数组长度
if (k == 0) return; // 如果 k 为 0,则无需旋转
reverse(nums, 0, numsSize - 1); // 逆置整个数组
reverse(nums, 0, k - 1); // 逆置前 k 个元素
reverse(nums, k, numsSize - 1); // 逆置剩余的元素
}
时间复杂度:O(N),空间复杂度:O(1)
方法2(使用临时数组):
cpp
void rotate(int* nums, int numsSize, int k) {
k = k % numsSize; // 防止 k 大于数组长度
if (k == 0) return; // 如果 k 为 0,则无需旋转
// 分配内存保存最后 k 个元素
int* temp = (int*)malloc(sizeof(int) * k);
if (temp == NULL) {
printf("动态开辟失败\n");
return;
}
// 保存最后 k 个元素到临时数组中
for (int i = 0; i < k; i++) {
temp[i] = nums[numsSize - k + i];
}
// 将前面的元素向后移动
for (int i = numsSize - k - 1; i >= 0; i--) {
nums[i + k] = nums[i];
}
// 将临时数组中的元素移到前面
for (int i = 0; i < k; i++) {
nums[i] = temp[i];
}
// 释放临时数组
free(temp);
}
时间复杂度:O(N) ,空间复杂度:O(N)