【两次过】Lintcode 1170. 重塑矩阵

在MATLAB中，有一个非常有用的函数叫做'reshape'，它可以将矩阵重新整形为一个不同大小的矩阵，但保留其原始数据。

先给出一个由二维数组表示的矩阵，以及两个正整数r和c，分别表示所需重新整形矩阵的行数和列数。

重新形成的矩阵需要用原始矩阵的所有元素以相同的行遍历顺序填充。

如果使用给定参数的“重塑”操作是可能且合法的，则输出新的重新整形矩阵; 否则，输出原始矩阵。

样例

样例1:

输入: 
nums = 
[[1,2],
 [3,4]]
r = 1, c = 4
输出: 
[[1,2,3,4]]
解释:
行遍历的顺序为 [1,2,3,4]. 新给出的矩阵大小为1 * 4, 根据前面给出的列表按行遍历即可。

样例2:

输入: 
nums = 
[[1,2],
 [3,4]]
r = 2, c = 4
输出: 
[[1,2],
 [3,4]]
解释:
无法将一个2 * 2的矩阵整形称为一个2 * 4的矩阵。 所以输出原始矩阵。

注意事项

矩阵行和列的大小范围为[1, 100]。
给出的r和c都为正数。

---

解题思路：

对于这种二维数组大小重新非配的问题的关键就是对应位置的坐标转换，最直接的办法就是先把原数组拉直，变成一条直线，然后再组成新的数组。直接就是遍历拉直后的一维数组的坐标，然后分别转换为两个二维数组的坐标进行赋值, 参见代码如下：

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 row = nums.length;
        int col = nums[0].length;
        if(r*c != row*col)
            return nums;
        
        int[][] res = new int[r][c];
        
        for(int i=0 ; i<r*c ; i++)
            res[i/c][i%c] = nums[i/col][i%col];

        
        return res;
    }
}