leetcode1260

class Solution:
    def shiftGrid(self, grid: List[List[int]], k: int) -> List[List[int]]:
        n = len(grid)#n行
        m = len(grid[0])#m列
        flat = [0] * (n * m)
        t = 0
        for i in range(n):
            for j in range(m):
                flat[t] = grid[i][j]
                t += 1
        k = k % (n * m)
        tail = flat[:n*m-k]
        head = flat[n*m-k:]
        flat = head + tail
        
        t = 0
        newgrid = [[0 for _ in range(m)]for _ in range(n)]
        for i in range(n):
            for j in range(m):
                newgrid[i][j] = flat[t]
                t += 1
        return newgrid

把二维数组 => 一维数组 => 一维数组变换 => 还原二维数组