给定一个数组,将数组中的元素向右移动 k 个位置,其中 k 是非负数。
进阶:
尽可能想出更多的解决方案,至少有三种不同的方法可以解决这个问题。
你可以使用空间复杂度为 O(1) 的 原地 算法解决这个问题吗?
示例 1:
输入: 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]
Solution 1 旋转数组
官方讲解的很清楚,这里稍微按凯话翻译一下
我们先把所有数组反转(逆序)
然后对前k个反转,这时候就相当于将后面k个移动到前面
此时再吧后n-k个反转,就变成了从头开始的数据
不得不说 秒啊
class Solution {
public:
void reverse(vector<int>& nums, int start, int end) {
while (start < end) {
swap(nums[start], nums[end]);
start += 1;
end -= 1;
}
}
void rotate(vector<int>& nums, int k) {
k %= nums.size();
reverse(nums, 0, nums.size() - 1);
reverse(nums, 0, k - 1);
reverse(nums, k, nums.size() - 1);
}
};
作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/rotate-array/solution/xuan-zhuan-shu-zu-by-leetcode-solution-nipk/
Solution 2 环状替换
因为是第i个位置的数放置到第(i+k)%n的位置上,所以我们可以通过一个循环,设置一个temp来放置患处的数,并记录下当前这个数的位置,通过循环一次所有数,就将所有的数换了位置,即每次一根据上一次的位置来改变。
但是!这时候当k是n的约数的时候,这玩意就开始跑循环了,所以我们得设置一个外循环来判断是否有重复出现,当又回到最初的起点的时候,我们就去下一个数继续循环改变位置。
class Solution {
public:
void rotate(vector<int>& nums, int k) {
int n=nums.size();
k=k%n;
int count = gcd(k, n);
for (int start = 0; start < count; ++start) {
int current = start;
int prev = nums[start];
do {
int next = (current + k) % n;
swap(nums[next], prev);
current = next;
} while (start != current);
}
}
};
总之看这种题,千万不要看下面的图,他的移动方式显然不是最好的,我们一般通过结果和最初的状态来想方法和思路。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/rotate-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。