189. 轮转数组【力扣(LeetCode) 】

最新推荐文章于 2024-10-05 12:29:44 发布

理论最高的吻

最新推荐文章于 2024-10-05 12:29:44 发布

阅读量316

点赞数 5

分类专栏：编程练习【LeetCode（力扣）】文章标签： leetcode 算法职场和发展 C++ 数学

本文链接：https://blog.csdn.net/yyh520025/article/details/140714765

版权

编程练习【LeetCode（力扣）】专栏收录该内容

72 篇文章 0 订阅

订阅专栏

一、题目描述

给定一个整数数组 nums，将数组中的元素向右轮转 k 个位置，其中 k 是非负数。

二、测试用例

示例 1:

输入: nums = [1,2,3,4,5,6,7], k = 3
输出: [5,6,7,1,2,3,4]
解释:
向右轮转 1 步: [7,1,2,3,4,5,6]
向右轮转 2 步: [6,7,1,2,3,4,5]
向右轮转 3 步: [5,6,7,1,2,3,4]

示例 2:

输入：nums = [-1,-100,3,99], k = 2
输出：[3,99,-1,-100]
解释: 
向右轮转 1 步: [99,-1,-100,3]
向右轮转 2 步: [3,99,-1,-100]

三、解题思路

基本思路：如果 k 确定，每个元素的最终位置也确定，可以直接将元素移到最终位置。
具体思路：
- 使用副本：建立原序列的副本，然后将元素移到最终位置，最终位置为 $(i+k)\%n$ 。【时间和空间复杂度都是 $\Omicron(n)$ 】
- 三次反转：记原序列为 abcde ，k = 2 。先将原序列反转，即 edcba ；在将前 k 个元素反转，即 decba ；在将后 n - k 个元素反转，即 deabc ，遂完成向右轮转 k 个位置。【空间复杂度为 $\Omicron(1)$ 】

四、参考代码

使用副本实现比较简单，故没有代码，下面是三次反转的代码：

时间复杂度： $\Omicron(n)$
空间复杂度： $\Omicron(1)$

void reverse(vector<int>& nums, int s, int e){
    int mid=(e+s)/2;
    int t;
    for(int i=s;i<=mid;i++){
        t=nums[i];
        nums[i]=nums[e-i+s];
        nums[e-i+s]=t;
    }
}
void rotate(vector<int>& nums, int k) {
    int n=nums.size();
    k=k%n;
    if(k==0) return ;
    reverse(nums,0,n-1);
    reverse(nums,0,k-1);
    reverse(nums,k,n-1);
}