题目
给你一个 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 次迁移操作后最终得到的 二维网格。
来源:力扣(LeetCode)
示例(示例 从leetcode直接抄的)
题意解析
根据原二维数组grid和移动次数k,得出移动后的二维数组,根据示例可以得出移动方式为每个数字向后移动k次,为一排最后一个数字时向下移动为下一排第一个,最后一排最后一个移动为第一排第一个。所以可以先获得grid的数组大小mn,移动次数为mn时结果是原数组,所以实际移动次数为k对m*n取模。
代码
public List<List<Integer>> shiftGrid(int[][] grid, int k) {
int m = grid[0].length;
int n = grid.length;
int index = k % (n * m);//实际移动次数
Integer[][] newGrid = new Integer[n][m];//结果数组
for (int i = 0; i < grid.length; i++) {
for (int i1 = 0; i1 < grid[0].length; i1++) {
int m1 = (i1 + index) % m;
int n1 = (i + (i1 + index) / m) % n;
newGrid[n1][m1] = grid[i][i1];//计算移动index次后的位置
}
}
List<List<Integer>> result= new ArrayList<>();//转换为List<List<Integer>>
for (int i = 0; i < m; i++) {
result.add(Arrays.asList(newGrid[i]));
}
return result;
}