(昨天没打卡一题,故今日补之)
题目
给定一个数组,将数组中的元素向右移动 k 个位置,其中 k 是非负数。
进阶:
尽可能想出更多的解决方案,至少有三种不同的方法可以解决这个问题。
你可以使用空间复杂度为 O(1) 的 原地 算法解决这个问题吗?
代码
class Solution {
public:
void rotate(vector<int>& array,int k) {//向右移动 k 个位置
int length = array.size();
k %= length;
reverse(array,0,length-1);
reverse(array,0,k-1);
reverse(array,k,length-1);
}
void reverse(vector<int>& array, int start,int end){
while(start < end){
int temp = array[start];
array[start] = array[end];
array[end] = temp;
start ++;
end--;
}
}
};
思路:
方法一:反转法
第一步:将所有元素反转
第二步:前 k 个元素反转(下标:0 ——> k-1)
第三步:剩下的元素反转(下标:k——> length -1)
如何反转数组呢?
你会发现其实就是 array [start ] 和 array [end] 交换位置
当然啦,一开始 start =0 ,end=length-1
然后start ++ ,end–
注意1:这会一直这样下去,条件是 start < end .
一开始我想的是 start !=end 但是这适用于奇数 1 2 3 4 5 start和end都指向3
但若是偶数呢 1 2 3 4 然后start指向3 end指向2,嘿又转回来了,所以我们的截止条件是
**start < end **
(注意:笔者说的都是下标 方便代码书写)
注意2:k %= length;
显而易见 倍数关系啊 ! 我们要消除掉。一开始忘记了 5555
方法二:临时数组
新开辟一个数组,然后再一一赋值给原先数组
class Solution {
public:
void rotate(vector<int>& array,int k) {//向右移动 k 个位置
int length = array.size();
vector<int> temp(length);
for (int i = 0; i < length; i++) {
temp[i] = array[i];
}
for (int i = 0; i < length; i++) {
array[(i + k) % length] = temp[i];
}
}
};
注意:
c++环境下报错
`
int temp[] = new int[length];
error: array initializer must be an initializer list
c++中创建一个数组
vector<int> temp(length);