【牛客题霸】旋转数组

题目描述

题目地址

思路

数组循环右移，我们可以看成先把数组整体翻转，再从前面翻转需要的位数。然后再把剩下的继续翻转，就可以得到答案

下面，我们来手动操作这一组数字

原始情况：1 2 3 4 5 6
整体翻转：6 5 4 3 2 1
从前翻转两位：5 6 4 3 2 1
把后四位翻转：5 6 1 2 3 4

这样是不是就与题目给定的返回值一样了呢？就是这么简单

题目中的坑

我们一定要注意，右移位数m可能会超过数组长度n，所以，一定要对m取模，得到正确的右移位数。

代码

import java.util.Arrays;
import java.util.Random;

/**
 * @author SQH
 */
public class RotateRight {
	//测试代码
	//因为比较简单就不用JUnit进行测试了
    public static void main(String[] args) {
        int[] arr = new int[10];
        Random random = new Random();
        for (int i = 0; i < arr.length; i++) {
            arr[i] = random.nextInt(100);
        }
        System.out.println("原始：" + Arrays.toString(arr));

        int[] solve = solve(arr.length, 2, arr);

        System.out.println("结果：" + Arrays.toString(solve));
    }

	//解题代码
    public static int[] solve(int n, int m, int[] a) {
        //右移距离可能超过数组长度
        m %= n;
        reverse(a, 0, n-1);
        reverse(a, 0, m-1);
        reverse(a, m, n - 1);

        return a;
    }

	//翻转数组
    public static void reverse(int[] arr, int left, int right) {
        while (left < right) {
            int temp = arr[left];
            arr[left++] = arr[right];
            arr[right--] = temp;
        }
    }
}