题目链接

Leetcode.189 轮转数组

题目描述

给你一个数组，将数组中的元素向右轮转 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]

提示：

$1<=nums.length<=10^5$
$-2^{31}<=nums[i]<=2^{31-1}$
$0<=k<=10^5$

分析：

可以通过以下三步，本地修改达到目的：

• 第一步将整个数组逆转
• 将 [ 0 , k - 1 ] 范围的元素逆转
• 将 [ k , n - 1 ] 范围的元素逆转

• 时间复杂度: $O(n)$

C++代码

class Solution {
public:
    void reverse(vector<int>& nums,int l,int r){
        for(int i = l,j = r;i < j;i++,j--){
            swap(nums[i],nums[j]);
        }
    }

    void rotate(vector<int>& nums, int k) {
       int n = nums.size();
       k %= n;
       reverse(nums,0,n-1);

       reverse(nums,0,k-1);
       reverse(nums,k,n-1);
    }
};

Java代码

class Solution {
    public void rotate(int[] nums, int k) {
        int n = nums.length;
        k = k % n;
        if(n<2||k==0) return;
        swap(0,n-1,nums);
        swap(0,k-1,nums);
        swap(k,n-1,nums);
    }

    public void swap(int i,int j,int[] nums){
        while(i<j){
            int tmp = nums[i];
            nums[i] = nums[j];
            nums[j] = tmp;
            i++;
            j--;
        }
    }
}