Create 1260-shift-2d-grid.py

This code is transcribed from the video https://www.youtube.com/watch?v=nJYFh4Dl-as

Author: AP-Repositories

python/1260-shift-2d-grid.py

@@ -0,0 +1,16 @@
+class Solution:
+    def shiftGrid(self, grid: List[List[int]], k: int) -> List[List[int]]:
+        M, N = len(grid), len(grid[0])
+        
+        def posToVal(r, c):
+            return r * N + c
+        def valToPos(v):
+            return [v // N, v % N] # r, c
+        
+        res = [[0] * N for i in range(M)]
+        for r in range(M):
+            for c in range(N):
+                newVal = (posToVal(r, c) + k) % (M * N)
+                newR, newC = valToPos(newVal)
+                res[newR][newC] = grid[r][c]
+        return res