OJ练习:1260. 二维网格迁移
题目要求:
给你一个 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 次迁移操作后最终得到的 二维网格。
解题思路:
双向队列 ,先将二维网格线性成一维的双向队列 执行k次操作:将双向队列的最后一个数字压入到队列的前面,弹出队列内的最后一个数字; 最后将一维的双向队列重新转换成二维的网格即可
代码如下:
class Solution {
public List<List<Integer>> shiftGrid(int[][] grid, int k) {
int[] nums = new int[grid.length * grid[0].length];
for(int i = 0, iMax = grid.length;i < iMax;i++) {
for(int j = 0, jMax = grid[0].length;j < jMax;j++) {
k %= nums.length;
nums[k++] = grid[i][j];
}
}
k = 0;
List<List<Integer>> lists = new ArrayList<>(grid.length);
for(int i = 0, iMax = grid.length;i < iMax;i++) {
List<Integer> tempList = new ArrayList<>(grid[0].length);
for(int j = 0, jMax = grid[0].length;j < jMax;j++) {
tempList.add(nums[k++]);
}
lists.add(tempList);
}
return lists;
}
}