Create 1260-shift-2d-grid.py

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

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