leetcode 周赛 163 5263. 二维网格迁移 python 解法

目录

1. 解法

给你一个 n 行 m 列的二维网格 grid 和一个整数 k。你需要将 grid 迁移 k 次。

每次「迁移」操作将会引发下述活动：

位于 grid[i][j] 的元素将会移动到 grid[i][j + 1]。
位于 grid[i][m - 1] 的元素将会移动到 grid[i + 1][0]。
位于 grid[n - 1][m - 1] 的元素将会移动到 grid[0][0]。
请你返回 k 次迁移操作后最终得到的 二维网格。

示例 1：

输入：grid = [[1,2,3],[4,5,6],[7,8,9]], k = 1
输出：[[9,1,2],[3,4,5],[6,7,8]]
示例 2：

输入：grid = [[3,8,1,9],[19,7,2,5],[4,6,11,10],[12,0,21,13]], k = 4
输出：[[12,0,21,13],[3,8,1,9],[19,7,2,5],[4,6,11,10]]
示例 3：

输入：grid = [[1,2,3],[4,5,6],[7,8,9]], k = 9
输出：[[1,2,3],[4,5,6],[7,8,9]]

提示：

1 <= grid.length <= 50
1 <= grid[i].length <= 50
-1000 <= grid[i][j] <= 1000
0 <= k <= 100

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/shift-2d-grid
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解法

class Solution:
    def shiftGrid(self, grid: List[List[int]], k: int) -> List[List[int]]:
        n = len(grid)
        m = len(grid[0])
        ans = [[0] * m for i in range(n)]
        for i in range(n):
            for j in range(m):
                l=(i*m+j+k)%(n*m);
                ans[l//m][l%m]=grid[i][j];
        return ans

这个是根据这次周赛第三名 JOHNKRAM 的代码写的。

原来的 grid[j][j] 会被换到 [(im+j+k)%(nm)//m][(i*m+j+k)%m] 的地方。

我的做法太笨了，我的方法就是直接模拟，c 记录了每列变换的次数。

class Solution:
    def shiftGrid(self, grid: List[List[int]], k: int) -> List[List[int]]:
        n = len(grid)
        m = len(grid[0])
        c = [0 for i in range(m)]
        x = m - 1
        for i in range(k):
            c[(x - i % m)] += 1
        x = -k % m
        r = []
        for i in range(m):
            col = []
            cn = (x + i) % m
            xx = -c[cn]
            for i in range(n):
                col.append(grid[(xx+i)%n][cn])
            r.append(col)
        for i in range(n):
            for j in range(m):
                grid[i][j] = r[j][i]
        return grid