【Java】LeetCode二维网格迁移

悠悠 2023-06-16 10:37 154阅读 0赞

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

示例 1：  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQzNDE2MjI2_size_16_color_FFFFFF_t_70]

输入：grid = \[\[1,2,3\],\[4,5,6\],\[7,8,9\]\], k = 1  
输出：\[\[9,1,2\],\[3,4,5\],\[6,7,8\]\]

package Leetcode;
    
    import java.util.ArrayList;
    import java.util.Arrays;
    import java.util.List;
    
    public class 二维网格迁移 {
        
        public static List<List<Integer>> shiftGrid(int[][] grid, int k) {
        
            int n=grid.length;
            int m=grid[0].length;
            while(k-->0) {
        
                int[] tmp = new int[n];
                for (int i = 0; i < n; i++)
                    tmp[i] = grid[i][0];
                int a = grid[n - 1][m - 1];
                for (int i = n - 1; i >= 1; i--)
                    grid[i][0] = grid[i - 1][m - 1];
                grid[0][0] = a;
                for (int i = 0; i < n; i++) {
        
                    for (int j = m - 1; j >= 1; j--) {
        
                        if (j == 1) grid[i][j] = tmp[i];
                        else grid[i][j] = grid[i][j - 1];
                    }
                }
            }
            List<List<Integer>> result = new ArrayList<List<Integer>>();
            for(int i=0;i<n;i++){
        
                List<Integer> list = new ArrayList<Integer>();
                for(int j=0;j<m;j++){
        
                    list.add(grid[i][j]);
                }
                result.add(list);
            }
            return result;
        }
    
        public static void main(String[] args) {
        
         int k=1;
         int [][]grid={
        {
        1,2,3},{
        4,5,6},{
        7,8,9}};
            System.out.println(shiftGrid(grid,k));
        }
    }

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQzNDE2MjI2_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/20191125222218742.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQzNDE2MjI2,size_16,color_FFFFFF,t_70