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次将最后一个元素移到第一个。
这样就会出现k大于vector的大小的情况,刚开始不理解题意,以为要将后面k个元素移到前面而已。
但当出现k大于vector的大小的时候,就不对了。
这时候取k%size就可以
先说这个题的核心想法吧:
拿把k元素从后面移到前面,可以分三步
(1)先把size-k个元素逆置
(2)再把后面k个元素逆置
(3)最后整体逆置
结果圈圈画画就可以看出来了
1 class Solution { 2 public: 3 void rotate(vector<int>& nums, int k) { 4 int s=nums.size(); 5 int c=0; 6 c=s-k%s; 7 reverse(nums.begin(),nums.begin()+c); 8 reverse(nums.begin()+c,nums.end()); 9 reverse(nums.begin(),nums.end()); 10 11 } 12 };