Go语言题解LeetCode1260二维网格迁移示例详解
刘09k11 人气:0题目描述
给你一个 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
思路分析
- 最笨的方法移动k次
- 第一次优化,对k取模,比如一个3*3的网格,移9次等于没有移
- 第二次优化,由k的值确定向左移简单还是向右移简单:正常情况是向右移,比如一个3*3的网格,要求移8次,相当于左移一次
- 第三次优化,空间换时间,需要多建立长度为k的临时数组,但这样可以把k次移动变成1次移动,除了需要移动到临时数组里的元素,其它元素的移动步长都是k。(这样一来,其实第二次优化的意义不是很大,就省了一点内存而已)
AC 代码
class Solution { public List<List<Integer>> shiftGrid(int[][] grid, int k) { int []yw=new int[grid.length*grid[0].length]; int p=0; for(int i=0;i<grid.length;i++) { for(int j=0;j<grid[0].length;j++) { yw[p++]=grid[i][j]; } } k=k%yw.length; if(k<yw.length/2) {//右移 int []tmp=new int[k]; int pnt=yw.length-k; for(int i=0;i<k;i++) { tmp[i]=yw[pnt]; pnt++; } for(int i=yw.length-k-1;i>=0;i--) { yw[i+k]=yw[i]; } for(int i=0;i<k;i++) { yw[i]=tmp[i]; } } else {//左移 k=yw.length-k; int []tmp=new int[k]; for(int i=0;i<k;i++) { tmp[i]=yw[i]; } for(int i=k;i<yw.length;i++) { yw[i-k]=yw[i]; } int pnt=yw.length-k; for(int i=0;i<k;i++) { yw[pnt]=tmp[i]; pnt++; } } List<List<Integer>> res=new ArrayList<>(); List<Integer> tp=new ArrayList<>(); p=0; for(int i=0;i<grid.length;i++) { for(int j=0;j<grid[0].length;j++) { tp.add(yw[p++]); } res.add(tp); tp=new ArrayList<>(); } return res; } }
加载全部内容