【Lintcode】1170. Reshape the Matrix

题目地址：

https://www.lintcode.com/problem/reshape-the-matrix/description

给定一个二维矩阵$A$，再给定两个正整数$r$和$c$，要求将$A$变换为$r$行$c$列的矩阵，并且保持行遍历相同。如果无法变换，则返回原矩阵，否则返回变换后的答案。

能变换当且仅当$A$的元素个数等于$rc$。判断完之后将原矩阵的数填到新矩阵里即可。代码如下：

public class Solution {
    /**
     * @param nums: List[List[int]]
     * @param r:    an integer
     * @param c:    an integer
     * @return: return List[List[int]]
     */
    public int[][] matrixReshape(int[][] nums, int r, int c) {
        // write your code here
        int m = nums.length, n = nums[0].length;
        if (m * n != r * c) {
            return nums;
        }
        
        int[][] res = new int[r][c];
        for (int idx = 0, i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                res[idx / c][idx % c] = nums[i][j];
                idx++;
            }
        }
        
        return res;
    }
}

时空复杂度$O(mn)$。