LeetCode #189 Rotate Array

来源:互联网 发布:windows管道通信 编辑:程序博客网 时间:2024/05/01 13:32

原题为:

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对象逆置。可以得到结果。


总结:解法一比较好理解,但是效率不如解法二。解法二比较难想到。


0 0