—给定一个数组,将数组中的元素向右移动 k 个位置,其中 k 是非负数。
—示例 1:
—输入: [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(1) 的 原地 算法。
c语言
- 暴力法
每次移动一个位置,移动K次,时间复杂度为O(n * k)
void rotate(int* nums, int numsSize, int k) {
for (int i = 0; i < k; i++) {
int pre = nums[numSize - 1];
for (int j = 0; j < numSize; j++) {
int tmp = pre;
pre = nums[j];
nums[j] = tmp;
}
}
}
- 反转法
这个方法基于这个事实:当我们旋转数组 k 次,k%n 个尾部元素会被移动到头部,剩下的元素会被向后移动。
在这个方法中,我们首先将所有元素反转。然后反转前 k 个元素,再反转后面n−k 个元素,就能得到想要的结果。
假设 n = 7且 k = 3; 时间复杂度为O(n)
| operation | change | | —- | —- | | 原始数组 | 1 2 3 4 5 6 7 | | 反转所有数字后 | 7 6 5 4 3 2 1 | | 反转前 k 个数字后 | 5 6 7 4 3 2 1 | | 反转后 n-k 个数字后 | 5 6 7 1 2 3 4 |
void reverse(int *nums, int start, int end);
void rotate(int* nums, int numsSize, int k) {
k %= numsSize;
reverse(nums, 0, numsSize -1);
reverse(nums, 0, k -1);
reverse(nums, k, numsSize - 1);
}
//反转就是向中间夹逼 交换头部和尾部
void reverse(int *nums, int start, int end) {
while (start < end) {
int tmp = nums[start];
nums[start] = nums[end];
nums[end] = tmp;
start++;
end--;
}
}
c plus
- 暴力法
和c语言暴力法相同,只不过把每次移动一次的for循环改成递归的形式class Solution {
public:
int tmp = 0;
void rotate(vector<int>& nums, int k) {
for(int i = 0; i < k; i++)
rotate_c(0, nums[0], nums);
}
void rotate_c(int i, int b, vector<int>& nums) {
if ((i + 1) == nums.size()) return;
tmp = nums[i + 1];
nums[i + 1] = b;
rotate_c(i + 1, tmp, nums);
nums[0] = tmp;
}
};
Go
append方法
用切片特性直接重新拼接,但空间复杂度不是O(1),append会生成一个新的切片,且内部会向移动数组一样移动每个数据,空间复杂度为O(n)func rotate(nums []int, k int) {
k %= len(nums)
copy(nums, append(nums[len(nums) - k: ], nums[ :len(nums) - k]...))
}
反转法
纯练习语法 ```go func rotate(nums []int, k int) { k %= len(nums) reverse(nums) reverse(nums[: k]) reverse(nums[k: ])
} func reverse(num []int) { i := 0 j := len(num) - 1 for i < j { num[i],num[j] = num[j],num[i] i++ j— } } ```