Go语言题解LeetCode1260二维网格迁移示例详解

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;
	}
}