知识点

利用翻转O(1)解决循环数组位移

思路

首先k对n取模,因为大于n的时候会循环回来,是没有意义的。

假如要把前k个数循环移动,我们可以先翻转前n-k个数,再翻转后k个数。

最后整个数组翻转即可实现。

时空复杂度

时间复杂度为O(n)

AC Code (C++)

class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param nums int整型vector 
     * @param k int整型 
     * @return int整型vector
     */
    vector<int> rotateCows(vector<int>& nums, int k) {
        int n = nums.size();
        k %= n;
        reverse(nums.begin(), nums.begin() + n - k);
        reverse(nums.begin() + n - k, nums.end());
        reverse(nums.begin(), nums.end());
        return nums;
    }
};