189. Rotate Array
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个元素不变序的搬到数组头部。
方法一:将后面k个元素逐个往前移动n-k
步,这种方式不适用额外的空间,但是提交后会超时。
方法二:适用STL的insert方法,逐个将后k个元素插入到头部,并将该元素从尾部删除
方法三:将整个序列翻转,然后分别翻转前面部分和后面部分
源码
方法一:
void rotate(vector<int>& nums, int k) {
// 将k之后的数和它前面的数逐个交换
int length = nums.size();
k = k % length;
int steps = length - k; // k和前面交换steps次就可以了
for(int i = (length - k); i < length; i++) {
int stop = i - steps; // 交换的截止位置
int tmp = nums[i];
int j = i;
while(j > stop) {
nums[j] = nums[j-1]; // 往后移动数字
j--;
}
nums[j] = tmp; // 将K数放到前面
}
}
方法二:
void rotate(vector<int>& nums, int k) {
for (int i = 0; i < k; ++i) {
nums.insert(nums.begin(), nums[nums.size() - 1]);
nums.pop_back();
}
}
方法三:
void rotate(vector<int>& nums, int k) {
int length = nums.size();
k = k % length; // 会出现k > length的情况
if(k == 0 || k >= length) return;
jk_reverse(nums,0,length);
jk_reverse(nums,0,k);
jk_reverse(nums,k,length - k);
}
void jk_reverse(vector<int>& nums, int start, int length) {
int middle = length / 2;
int i = 0;
while(i < middle) {
swap(nums[start + i], nums[start + (length - i - 1)]);
i++;
}
}
void swap(int &a, int &b) {
a ^= b;
b ^= a;
a ^= b;
}