Leetcode 1386 Solution

This article provides solution to leetcode question 1386 (shift-2d-grid)
Categories:
Tags:
Access this page by simply typing in "lcs 1386" in your browser address bar if you have bunnylol configured.

https://leetcode.com/problems/shift-2d-grid

Solution

class Solution:
    def shiftGrid(self, grid: List[List[int]], k: int) -> List[List[int]]:
        m = len(grid)
        n = len(grid[0])

        ans = [[0 for _ in range(n)] for _ in range(m)]

        for i in range(m):
            for j in range(n):
                index = ((i * n + j) + k) % (m * n)

                r = index // n
                s = index % n
                ans[r][s] = grid[i][j]

        return ans