原题为:
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]
.
一开始想着建立额外的数组,直接将数组中的对象定位到新的位置,但后来发现比较困难,原因在于情况很多,并不是固定的。后来发现每走一步,数组尾部的值会移到头部,因此得到解法一:
void rotate1(int nums[], int n, int k) {
if(k <= 0)
return;
k %= n; //N步一循环
while( k > 0 )
{
int tmp = nums[n - 1]; //每次移动末尾数字到头
for(int i = n - 1; i > 0; i--)
nums[i] = nums[i - 1];
nums[0] = tmp;
k--;
}
}
};
但是此解法提交后会超时。 改进方法(我称之为三步法):
void rotate(int nums[], int n, int k)
{
reverse(nums,nums+n);
reverse(nums,nums+k%n);
reverse(nums+k%n,nums+n);
}
解释:第一步将整个数组逆置;第二步将0 到k-1对象逆置;第三步将k到n-1对象逆置。可以得到结果。
总结:解法一比较好理解,但是效率不如解法二。解法二比较难想到。