Задача: 1260. Shift 2D Grid
Сложность: easy

Дана двумерная сетка размером m x n и целое число k. Требуется сдвинуть сетку k раз. За одну операцию сдвига: элемент в grid[i][j] перемещается в grid[i][j + 1]. Элемент в grid[i][n - 1] перемещается в grid[i + 1][0]. Элемент в grid[m - 1][n - 1] перемещается в grid[0][0]. Верните двумерную сетку после применения операции сдвига k раз.

Пример:

Input: grid = [[1,2,3],[4,5,6],[7,8,9]], k = 1
Output: [[9,1,2],[3,4,5],[6,7,8]]

👨‍💻 Алгоритм:

1⃣Преобразовать двумерную сетку в одномерный массив.

2⃣Выполнить сдвиг элементов в одномерном массиве.

3⃣Преобразовать одномерный массив обратно в двумерную сетку.

😎 Решение:

class Solution {
    fun shiftGrid(grid: Array<Array<Int>>, k: Int): Array<Array<Int>> {
        val m = grid.size
        val n = grid[0].size
        val total = m * n
        var k = k % total

        if (k == 0) {
            return grid
        }

        val flatArray = IntArray(total)
        for (i in 0 until total) {
            flatArray[i] = grid[i / n][i % n]
        }

        val newArray = IntArray(total)
        for (i in 0 until total) {
            newArray[(i + k) % total] = flatArray[i]
        }

        val newGrid = Array(m) { Array(n) { 0 } }
        for (i in 0 until m) {
            for (j in 0 until n) {
                newGrid[i][j] = newArray[i * n + j]
            }
        }

        return newGrid
    }
}

Ставь 👍 и забирай 📚 Базу знаний