给定一个数组,将数组向右移动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]
class Solution {
public:
/**
* @param nums: an array
* @param k: an integer
* @return: rotate the array to the right by k steps
*/
vector<int> rotate(vector<int> &nums, int k) {
// Write your code here
int array[k];
int len=nums.size();
if(k>len) k%=len;//翻转一圈仍为该数组,取余才为该翻转次数
for (int i = nums.size()-k,j=0; i < nums.size(); i++,j++) {//记录后面k个数值
/* code */
array[j]=nums[i];
}
for (int i = nums.size(); i >= k; i--) {//从后面开始遍历,替代
/* code */
nums[i]=nums[i-k];
}
for (int i = 0; i < k; i++) {//将最前面k个值替代
/* code */
nums[i]=array[i];
}
return nums;
}
};