【Leetcode】1260. Shift 2D Grid

题目地址：

https://leetcode.com/problems/shift-2d-grid/

给定一个$m\times n$的矩阵，再给定一个非负整数$k$，要求将矩阵按行展开为数组之后再向右顺次移动$k$位（出界的补到左边），再还原为$m\times n$矩阵。求最后所得的矩阵。

先将$k$对$mn$取余，然后找到最后矩阵的第一个数字所在位置，从这个位置开始向后添加，然后再从头开始添加直到这个位置。代码如下：

import java.util.ArrayList;
import java.util.List;

public class Solution {
    public List<List<Integer>> shiftGrid(int[][] grid, int k) {
        int m = grid.length, n = grid[0].length;
        k %= m * n;
        int x = 0, y = 0;
        label:
        for (int i = m - 1; i >= 0; i--) {
            for (int j = n - 1; j >= 0; j--) {
                k--;
                
                if (k == 0) {
                    x = i;
                    y = j;
                    break label;
                }
            }
        }
        
        List<List<Integer>> res = new ArrayList<>();
        for (int i = x; i < m; i++) {
            int j = i == x ? y : 0;
            for (; j < n; j++) {
                add(res, n, grid[i][j]);
            }
        }
        for (int i = 0; i <= x; i++) {
            int bound = i == x ? y : n;
            for (int j = 0; j < bound; j++) {
                add(res, n, grid[i][j]);
            }
        }
        
        return res;
    }
    
    private void add(List<List<Integer>> list, int n, int x) {
        if (list.isEmpty() || list.get(list.size() - 1).size() == n) {
            list.add(new ArrayList<>());
        }
        
        list.get(list.size() - 1).add(x);
    }
}

时间复杂度$O(mn)$，空间$O(1)$。