JavaМоим ученикам

Циклический сдвиг в массиве

Не так давно стартовал очередной курс Java на одном небезызвестном образовательном портале. И вот, моим студентам досталась задача по работе с массивами. Статья в первую очередь для них, но и для интересующихся, конечно же 🙂 Отдельное спасибо alexandr.baykov@gmail.com за комментарий по поводу массива с чётным количеством элементов и чётным размером сдвига. Я переписал алгоритм и обновил статью.

Итак, задача сформулирована следующим образом

Написать метод, которому на вход подается одномерный массив и число n (может быть положительным, или отрицательным), при этом метод должен сместить все элементы массива на n позиций. Для усложнения задачи нельзя пользоваться вспомогательными массивами.

Усложняющее условие было введено потому, что можно решить задачу при помощи разделения массивов. В таком случае решение состоит в том, чтобы «отрезать» от массива кусок длиной n справа или слева (в зависимости от того, как n сравнивается с 0) и «прикрепить» отрезанную часть обратно с другой стороны. Это довольно дешёвый алгоритм, который имеет сложность O(n), но он будет требовать дополнительной памяти для хранения временного массива размера n. Поскольку такое решение самое простое, да и не особенно подходит под условие задачи, то мы перейдём сразу к более сложной алгоритмизации.

Разумеется, после первых подходов многие студенты находят следующее довольно логичное и вполне корректное решение. Оно весьма популярно на различных сайтах, описывающих циклический сдвиг. Суть его в том, что любой сдвиг можно представить в качестве n сдвигов на 1 ячейку. Сдвиг на 1 ячейку довольно просто реализуется циклом. В итоге, если изобразить это Java-методом, получается примерно такой метод:


public static int[] shiftArray(int[] incomingArray, int shift) {
    if(shift != 0){
    // Любой сдвиг больше длины массива можно оптимизировать до меньшего сдвига
    // через деление по модулю

    int finalShift;
    if (shift > incomingArray.length) {
        shift = Math.abs(shift % incomingArray.length);
    }
    else {
        finalShift = shift;
    }

    // в зависимости от знака сдвига движение будет происходить
    // слева направо при положительном сдвиге
    // справа налево при отрицательном
    if (shift > 0) {
        for (int n = 0; n < shift; n++) {
        // убираем первый элемент в буфер, а на его место ставим хвостовой элемент
        int buffer = incomingArray[0];
        incomingArray[0] = incomingArray[incomingArray.length - 1];
       
        // циклично сдвигаем весь массив
        for (int j = 1; j < incomingArray.length - 1; j++) {
           incomingArray[incomingArray.length - j] = incomingArray[incomingArray.length - j - 1]; }
            // ставим буферный элемент в 1 ячейку
            incomingArray[1] = buffer;
        }
    }
    else if (shift < 0) {
        for (int i = 0; i > shift; i--) {
            int buffer = incomingArray[incomingArray.length - 1];
            incomingArray[incomingArray.length - 1] = incomingArray[0];

            for (int j = 1; j < incomingArray.length - 1; j++) {
                 incomingArray[j - 1] = incomingArray[j];
            }
         incomingArray[incomingArray.length - 2] = buffer;
         }
      }
    }
    return incomingArray;
}

Хочу отметить, что такое решение имеет место, и оно применимо. Но мы же решаем алгоритмическую задачу, поэтому неплохо бы поговорить о том, какую сложность будет иметь приведенный алгоритм. Если опустить процесс вычислений буферных элементов и обмена ими между ячейками, то сложность алгоритма сводится к O(N * M), где N — размер входного массива, а M — величина сдвига. Очевидно, что для |M| = 1 (т.е. M = -1 или M = 1) сложность снизится до O(N). Это частный случай.

Теперь давайте подумаем, что же тут не так. На самом деле, если мы знаем величину сдвига (а мы её знаем), то вполне можно вычислить, какой элемент будет следующим. Это говорит о том, что можно создать жонглирующий алгоритм следующего вида:

  1. Получаем нулевой элемент и кладем его в буфер
  2. Начинаем цикл от 0 до длины массива включительно (это обусловлено тем, что в момент, когда мы дойдём до последнего элемента, он будет помещен в буфер, из которого его надо восстановить на месте нулевого элемента, чем полностью закольцевать сдвиг)
  3. Дальше мы меняем местами элементы, исходя из размера шага. Например, для шага 3 и массива длиной 7 мы сначала поменяем элементы 0, 3, 6. Затем — 1, 4, 2. И так далее.

Если рассмотреть наборы изменяемых элементов, то мы увидим, что количество таких наборов, внутри которых мы будем итеративно менять местами элементы, будет равно наименьшему общему делителю для размера массива и размера сдвига. К примеру, для массива из 12 элементов при сдвиге 3 мы будем иметь три набора смещаемых элементов


 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12

- 1, 4, 7, 10
- 2, 5, 8, 11
- 3, 6, 9, 12

4 5 6 7 8 9 10 11 12 1 2 3

Поскольку эти наборы независимы, то при перестановке в них элементов местами мы не нарушим общий порядок следования элементов.

В реализации нужно учитывать и то, что сдвиг может идти как влево, так и вправо.

Теперь мы можем написать реализацию нашего алгоритма


public class moves {

    public static void main(String[] args){
        int[] in1 = {5, 3, 7, 2, 9, 13, 42};
        int[] out1 = moves(in1, -2);
        printOut(out1);

        int[] in2 = {5, 3, 7, 2, 9, 13, 42, 43, 45, 46};
        int[] out2 = moves(in2, 2);
        printOut(out2);

        int[] in3 = {5, 3, 7, 2, 9, 13, 42, 143};
        int[] out3 = moves(in3, 5);
        printOut(out3);

        int[] in4 = {5, 3, 7, 2, 9, 42, 43, 45, 46};
        int[] out4 = moves(in4, 0);
        printOut(out4);
    }

    /**
     * Main method, shifts array
     * @param incoming - user array
     * @param delta - shift value
     * @return int[] result array
     */
     static int[] moves(int[] incoming, int delta){
         int currentIndex, movedIndex, buffer;
         for (int i = 0; i < greatestCommonDivisor(delta, incoming.length); i++) {
             buffer = incoming[i];
             currentIndex = i;

             if(delta > 0){
                 while (true) {
                     movedIndex = currentIndex + delta;
                     if (movedIndex >= incoming.length)
                         movedIndex = movedIndex - incoming.length;

                     if (movedIndex == i)
                         break;

                     incoming[currentIndex] = incoming[movedIndex];
                     currentIndex = movedIndex;
                 }
             }
             else if(delta < 0){
                 while (true) {
                     movedIndex = currentIndex + delta;
                     if (movedIndex < 0)
                         movedIndex = incoming.length + movedIndex;
                 
                     if (movedIndex == i)
                         break;

                     incoming[currentIndex] = incoming[movedIndex];
                     currentIndex = movedIndex;
                 }
             }

        incoming[currentIndex] = buffer;
        }

    return incoming;
    }

    /**
    * Simple printout
    * @param incoming Array user array
    */
    public static void printOut(int[] incomingArray){
        for(int item: incomingArray){
            System.out.print(item + " ");
        }

        System.out.println();
    }

      /**
      * Finding the GCD in recoursive function
      * @param a - first element
      * @param b - second element
      * @return int GCD
      */
     static int greatestCommonDivisor(int a, int b)
     {
         if (b == 0)
             return a;
         else
             return greatestCommonDivisor(b, a % b);
     }
 }

Как и в прошлый раз, мы можем пренебречь расчетами индексов. И в конечном итоге мы получаем сложность алгоритма, которая зависит исключительно от длины массива, т.е. O(N). И эта сложность не будет меняться в зависимости от исключительных ситуаций, что как минимум не хуже предыдущего алгоритма, а для сдвигов, больших, чем единица, лучше.

Надеюсь, приведенные измышления будут для Вас полезны. Буду рад критике и комментариям!

Дочитавшим до конца, картинка демотиватор 🙂

Циклический сдвиг в массиве: 3 комментария

  1. Данное решение не работает, если длина массива четная и delta тоже четное число.
    На вход подан массив
    int[] in = {5, 3, 7, 2, 9, 13, 42, 43, 45, 46}
    int[] out3 = moves(in, 6);
    printOut(out3); — выводит 45 3 5 2 7 13 5 43 42 46

    если delta = 2, вывод — 42 3 5 2 5 13 7 43 9 46

  2. Надо избавиться от возможных отрицательных результатов метода greatestCommonDivisor (например для пары 6, -1 и т.п.).

Добавить комментарий

Ваш адрес email не будет опубликован. Обязательные поля помечены *