知识点
利用翻转O(1)解决循环数组位移
思路
首先k对n取模,因为大于n的时候会循环回来,是没有意义的。
假如要把前k个数循环移动,我们可以先翻转前n-k个数,再翻转后k个数。
最后整个数组翻转即可实现。
时空复杂度
时间复杂度为
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; } };