1260. 二维网格迁移-蒲公英云

1260. 二维网格迁移

给你一个 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 次迁移操作后最终得到的 二维网格。

示例 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]]

提示：

m == grid.length
n == grid[i].length
1 <= m <= 50
1 <= n <= 50
-1000 <= grid[i][j] <= 1000

0 <= k <= 100

package Solution1260;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;

class Solution {

public List<List<Integer>> shiftGrid(int[][] grid, int k) {
    int n = grid.length * grid[0].length;
    List<List<Integer>> out = new ArrayList<List<Integer>>();
    Queue<Integer> q1 = new LinkedList<>();
    Queue<Integer> q2 = new LinkedList<>();
    k = k % (grid.length * grid[0].length);
    int index = 0;
    for (int i = 0; i < grid.length; i++) {
        for (int j = 0; j < grid[0].length; j++) {
            index = i * grid[0].length + j;

// System.out.println(i * grid[0].length + j);

            if (n - index <= k) {
                q1.add(grid[i][j]);
            }
            if (n - index > k) {
                q2.add(grid[i][j]);
            }
        }
    }

// System.out.println(q1);
// System.out.println(q2);

    for (int i = 0; i < grid.length; i++) {
        List<Integer> row = new ArrayList<Integer>();
        for (int j = 0; j < grid[0].length; j++) {
            if (q1.size() > 0) {
                row.add(q1.remove());
            } else {
                row.add(q2.remove());
            }
        }
        out.add(row);
    }
    return out;
}
public static void main(String[] args) {
    Solution sol = new Solution();
    int[][] grid = { { 3, 8, 1, 9 }, { 19, 7, 2, 5 }, { 4, 6, 11, 10 }, { 12, 0, 21, 13 } };
    int k = 4;
    System.out.println(sol.shiftGrid(grid, k));
}