189.轮转数组

一个简单的题目但是想要做出空间复杂度O(1)的做法还是有难度的

先说说简单的做法

1.让nums每次整体右移一位,移动k次 时间O(nk) 空间O(1)

2.使用一个标记数组 每次移动k位 时间O(n) 空间O(n)

再来看看优秀的做法

方法一: 环状替换法

我们直接把一个元素替换到它最终要去的地方(i+k)%n ,但是这样可能会导致那个地方的数值被覆盖,所以我们使用tmp 和nextTmp记录这两个地方的数值,在更新中迭代

复制代码
     int tmp=nums[curIndex],nextTmp=nums[(curIndex + k)%len];
            while(true){
                nums[(curIndex + k)%len] = tmp;
                tmp = nextTmp;
                curIndex = (curIndex + k) % len;
                if(curIndex == curTime) break;
                nextTmp = nums[(curIndex + k)%len];
            }

但是这样的循环要走几次呢 , 我们假设从起点出发,一步跨过k个元素,最后回到起点走了a圈,那么an=kb,这里的b是走一趟会遇到的元素, ++注意这个等式就说明an是n和k的公倍数了,因为它除以n或者k得到的结果都是整数.++ 又因为我们是第一次回到起点,这说明a是最小的, 所以an是n和k的最小公倍数 lcm(n,k) 所以b=lcm(n,k)/k 那么一共要走 n/b =nk/lcm(n,k) = gcd(n,k) 因为最小公倍数和最大公约数的乘积就是他们本身

复制代码
class Solution {
public:
    void rotate(vector<int>& nums, int k) {
        int len = nums.size();
        int times = 1;
        if(k==0) return;
        times = gcd(len,k);
        for(int curTime = 0;curTime<times;curTime++){
            int curIndex = curTime;
            int tmp=nums[curIndex],nextTmp=nums[(curIndex + k)%len];
            while(true){
                nums[(curIndex + k)%len] = tmp;
                tmp = nextTmp;
                curIndex = (curIndex + k) % len;
                if(curIndex == curTime) break;
                nextTmp = nums[(curIndex + k)%len];
            }
        }
    }
};

方法二: 数组反转法

不妨取 k%=nums.size(),因为每走一个nums.size() 数组其实又回到了起点

之后(0<=k<nums.size())我们来看 题目要求的结果中 ++数组最后的k个元素会被移动到数组最前面,而我们知道 一个序列反转两次就会变成原来的顺序++

所以我们可以先反转一次使得最后的k个元素到首部 ,之后对[0,k mod n−1] 区间的元素和 [k mod n,n−1] 分别再反转一次就得到答案.

复制代码
class Solution {
public:
    void rotate(vector<int>& nums, int k) {
        k%=nums.size();
        reverse(nums.begin(),nums.end());
        reverse(nums.begin(),nums.begin()+k);
        reverse(nums.begin()+k,nums.end());
    }
};
相关推荐
NAGNIP9 小时前
大模型框架性能优化策略:延迟、吞吐量与成本权衡
算法
美团技术团队10 小时前
LongCat-Flash:如何使用 SGLang 部署美团 Agentic 模型
人工智能·算法
Fanxt_Ja15 小时前
【LeetCode】算法详解#15 ---环形链表II
数据结构·算法·leetcode·链表
侃侃_天下15 小时前
最终的信号类
开发语言·c++·算法
茉莉玫瑰花茶15 小时前
算法 --- 字符串
算法
博笙困了15 小时前
AcWing学习——差分
c++·算法
NAGNIP15 小时前
认识 Unsloth 框架:大模型高效微调的利器
算法
NAGNIP15 小时前
大模型微调框架之LLaMA Factory
算法
echoarts15 小时前
Rayon Rust中的数据并行库入门教程
开发语言·其他·算法·rust
Python技术极客15 小时前
一款超好用的 Python 交互式可视化工具,强烈推荐~
算法