题目链接:初级算法 - LeetBook - 力扣(LeetCode)全球极客挚爱的技术成长平台
思路一:第一想法就是利用辅助数组,得到平移后的结果,再覆盖回原数组,并且数组平移运算可以采用取模操作。
class Solution {
public:
void rotate(vector<int>& nums, int k) {
int len = nums.size();
vector<int> tmp(len);
for (int i = 0; i < len; i++) {
tmp[(i + k) % len] = nums[i];
}
for (int i = 0; i < len; i++) {
nums[i] = tmp[i];
}
}
};
思路二:题目中提示可以采用原地旋转的方式,可以考虑用翻转数组代替平移操作,向右循环平移 个单位,即相当于翻转数组后,再将区间
和
分别进行翻转。
class Solution {
public:
void rotate(vector<int>& nums, int k) {
k = k % nums.size();
reverse(nums.begin(), nums.end());
reverse(nums.begin(), nums.begin() + k);
reverse(nums.begin() + k, nums.end());
}
};
解题思路: 看完题解后,发现还有另外一种解法,也可以实现原地旋转,要用到一定的数学原理,有点难思考,下面给出代码,数学推导可以参考旋转数组 - 轮转数组 - 力扣(LeetCode)。
class Solution {
public:
void rotate(vector<int>& nums, int k) {
int n = nums.size();
k = k % n;
int cnt = __gcd(k, n); // __gcd()在头文件algorithm中,求最大公约数
for (int i = 0; i < cnt; i++) {
int cur = i;
int pre = nums[i];
do {
int next = (cur + k) % n;
swap(nums[next], pre);
cur = next;
} while(i != cur);
}
}
};