原题链接

这道题的思想非常巧妙,先将整个数组反转,然后反转 [0, k) 前 k 个数,再反转第 k 个数以后的数 [k, n),就完成了数组元素整体右移 k 个位置。

1
2
3
4
5
6
7
8
9
10
class Solution {
public:
void rotate(vector<int>& nums, int k) {
int n = nums.size();
k%=n;
reverse(nums.begin(), nums.end());
reverse(nums.begin(), nums.begin() + k);
reverse(nums.begin()+k, nums.end());
}
};