leetcode1260二维网络迁移

题目

n*m的矩阵,一个整数k,移动矩阵k次。每次移动的操作为:

  • 向右移动(最后一列移动到第一列)
  • 之后,第一列向下移动。

1<=N、M<=50
0<=k<=100

题解

思考:
从总体思考:相当于向右移动k%m列;另mod=k%m,donw=k/m。则如果mod为0,则所有列都下移donw行。否则,前mod列移动donw+1行,后面的向下移动donw行。

代码上实现如下:向右tmp[i][(j+k)%grid[0].length]=grid[i][j];,向下grid[(i+downTimes)%grid.length][j]=tmp[i][j];
这样避免实际根据k一个一个移动,时间复杂度为O(N*M)。

public List<List<Integer>> shiftGrid(int[][] grid, int k) {
    int[][] tmp=new int[grid.length][grid[0].length];
    for (int i = 0; i < grid.length; i++) {//每行都向右移动k个单位
        for (int j = 0; j < grid[0].length; j++) {
            tmp[i][(j+k)%grid[0].length]=grid[i][j];
        }
    }
    //向下的列数
    int donw=k/grid[0].length;
    int mod=k%grid[0].length;
    for (int i = 0; i < grid.length; i++) {
        for (int j = 0; j < grid[0].length; j++) {
            int downTimes=donw;
            if(j<mod){
                downTimes++;
            }
            grid[(i+downTimes)%grid.length][j]=tmp[i][j];
        }
    }
    //转化为输出要求的格式
    List<List<Integer>> res=new ArrayList<>(grid.length);
    for (int i = 0; i < grid.length; i++) {
        List<Integer> list=new ArrayList<>(grid[0].length);
        for (int j = 0; j < grid[0].length; j++) {
            list.add(grid[i][j]);
        }
        res.add(list);
    }
    return res;
}
原文地址:https://www.cnblogs.com/FannyChung/p/leetcode1260.html