LeetCode 初级 - 旋转数组

旋转数组

给定一个数组，将数组中的元素向右移动 k 个位置，其中 k 是非负数。

示例 1:

输入: [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:

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

说明:

• 尽可能想出更多的解决方案，至少有三种不同的方法可以解决这个问题。
• 要求使用空间复杂度为 O(1) 的原地算法。

分析

只想了两种解决方法：

第一种解决方法：循环k次，每次向后移动一位；

第二种解决方法：经过3次子数组的 “对称翻折”：

如 [1,2,3,4,5,6,7] 和 k = 3，第一步[1,2,3,4]->[4,3,2,1]，第二步[5,6,7]->[7,6,5]，最后[4,3,2,1,7,6,5]->[5,6,7,1,2,3,4]

代码

实现第二种解法：

class Solution {
        public void rotate(int[] nums, int k) {
            int count = k%nums.length;
            int limit = nums.length - count;

            reverse(nums,0,limit-1);
            reverse(nums,limit,nums.length-1);
            reverse(nums,0,nums.length-1);


        }
        //子数组对称
        private void reverse(int[] arr,int st,int end){
                while( st < end){
                    int temp = arr[st];
                    arr[st] = arr[end];
                    arr[end] = temp;
                    st ++;
                    end --;
                }
            }
    }