LeetCode189 旋转数组

发表于 | 分类于 | 阅读次数：

题目描述

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

实例1：

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

实例2：

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

解决方案

直接遍历移动

算法思想：循环 k 次，每次循环将数组中的数据向后移动一位，最后一位移动到第一位。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
class Solution {
    public void rotate(int[] nums, int k) {
        if(nums==null||nums.length==0) return;
        k = k % nums.length;
        while(k>0){
            int temp = nums[nums.length-1];
            for(int i=nums.length-1;i>0;i--){
                nums[i] = nums[i-1];
            }
            nums[0] = temp;
            k--;
        }
    }
}

一次移动到相应的位置

算法思想：将第 i 位数据直接移动到第 i + k 位。算法总共移动次数为 N - 1 次。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
class Solution {
    public void rotate(int[] nums, int k) {
        k = k % nums.length;
        int count = 0;
        for (int start = 0; count < nums.length; start++) {
            int current = start;
            int prev = nums[start];
            do {
                int next = (current + k) % nums.length;
                int temp = nums[next];
                nums[next] = prev;
                prev = temp;
                current = next;
                count++;
            } while (start != current);
        }
    }
}

三次反转

算法思想：第一次将所有元素进行反转，第二次将前 k 个元素进行反转，第三次将后 N - k 个元素进行反转。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
class Solution {
    public void rotate(int[] nums, int k) {
        k %= nums.length;
        reverse(nums,0,nums.length-1);
        reverse(nums,0,k-1);
        reverse(nums,k,nums.length-1);
    }
    public int[] reverse (int[] nums,int start,int end){
        int temp;
        while(start<end){
            temp=nums[start];
            nums[start]=nums[end];
            nums[end]=temp;
            start++;
            end--;
        }
        return nums;
    }
}