给你一个 m 行 n 列的二维网格 grid 和一个整数 k。你需要将 grid 迁移 k 次。
每次「迁移」操作将会引发下述活动:
位于 grid[i][j] 的元素将会移动到 grid[i][j + 1]。 位于 grid[i][n - 1] 的元素将会移动到 grid[i + 1][0]。 位于 grid[m - 1][n - 1] 的元素将会移动到 grid[0][0]。 请你返回 k 次迁移操作后最终得到的 二维网格。
class Solution {
public List<List<Integer>> shiftGrid(int[][] grid, int k) {
List<List<Integer>> result = new ArrayList<>();
int m = grid.length;
int n = grid[0].length;
for (int i = 0; i < m; i++) {
List<Integer> r = new ArrayList<>();
for (int j = 0; j < n; j++) {
r.add(0);
}
result.add(r);
}
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
int index = (i * n + j + k) % (m * n);
result.get(index / n).set(index % n, grid[i][j]);
}
}
return result;
}
}
每次移动都相当于一个一维的列表向后推移一位。解决的思想是把多维数组压成一个一行的,根据移动的次数,可以得到新的索引。最后再把新的一行的数据通过计算放回结果。