二维网格迁移

/*
 * 思路：
 * 1. 使用流将二维数组展平为一维数组。
 * 2. 通过计算迁移次数k得到最终的位置。
 * 3. 使用流将一维数组重新构造成二维数组。
 */
import java.util.Arrays;
import java.util.stream.IntStream;

public class Solution {
    public int[][] shiftGrid(int[][] grid, int k) {
        int m = grid.length;
        int n = grid[0].length;
        int[] flat = Arrays.stream(grid).flatMapToInt(Arrays::stream).toArray();
        int[] shifted = new int[m * n];
        IntStream.range(0, m * n).forEach(i -> shifted[(i + k) % (m * n)] = flat[i]);
        return IntStream.range(0, m).mapToObj(i -> Arrays.copyOfRange(shifted, i * n, (i + 1) * n)).toArray(int[][]::new);
    }
}