【leetcode面试经典 150 题】189. 轮转数组

题目

给定一个整数数组 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]

提示：

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

代码

总共反转三次，两边分别反转，整体再转一次即可。

class Solution {
    public int[] totateTemp(int[] nums, int a, int b) {
        for (int i = 0; i < (b - a + 1) / 2; i++) {
            int temp = nums[i + a];
            nums[i + a] = nums[b - i];
            nums[b - i] = temp;
        }
        return nums;
    }

    public void rotate(int[] nums, int k) {
        int n = nums.length;
        k = k % n;
        nums = totateTemp(nums, 0, n - k - 1);
        nums = totateTemp(nums, n - k, n - 1);
        nums = totateTemp(nums, 0, n - 1);
    }
}