1. 题目来源
链接:旋转数组
来源:LeetCode
2. 题目说明
给定一个数组,将数组中的元素向右移动 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]
示例2:
输入: [-1,-100,3,99] 和 k = 2
输出: [3,99,-1,-100]
解释:
向右旋转 1 步: [99,-1,-100,3]
向右旋转 2 步: [3,99,-1,-100]
3. 题目解析
此题可以采用头插法,一个一个的移动。但是有种更加简单的选择数组的方式。可以采用翻转的方式,比如12345经过翻转就变成了54321,这样已经做到了把前面的数字放到后面去,但是还没有完全达到要求,比如,只需要把12放在后面去,目标数组就是34512,与54321对比发现我们就只需要在把分界线前后数组再进行翻转一次就可得到目标数组了。所以此题只需要采取三次翻转的方式就可以得到目标数组,首先翻转分界线前后数组,再整体翻转一次即可。此题面试常考,大家可以记一下此方法。
- 三重反转
把一个数组的右边一部分移动到左边相当于:
把左部分翻转
把右部分翻转
最后把整体翻转
或是,直接调用算法头文件< algrorithm>中的函数rotate,函数原型可在CSDN上自行查找相关内容,最好能看看STL源码,或是《STL源码剖析》侯捷著。
4. 代码展示
class Solution {
public:
void rotate(vector<int>& nums, int k) {
// std::rotate(nums.begin(), nums.end() - k % nums.size(), nums.end());
reverse(nums.begin(), nums.end() - k % nums.size());
reverse(nums.end() - k % nums.size(), nums.end());
reverse(nums.begin(), nums.end());
}
};