LeetCode-566

重塑矩阵

题目
在 MATLAB 中，有一个非常有用的函数 reshape ，它可以将一个 m x n 矩阵重塑为另一个大小不同（r x c）的新矩阵，但保留其原始数据。
给一个由二维数组 mat 表示的 m x n 矩阵，以及两个正整数 r 和 c ，分别表示想要的重构的矩阵的行数和列数。
重构后的矩阵需要将原始矩阵的所有元素以相同的行遍历顺序填充。
如果具有给定参数的 reshape 操作是可行且合理的，则输出新的重塑矩阵；否则，输出原始矩阵。

示例
输入：mat = [[1,2],[3,4]], r = 1, c = 4
输出：[[1,2,3,4]]

题解

关于两个二维数组的映射，可以先将二维数组映射为一个一维数组，再将这个一维数组映射回 r 行 c 列的二维数组。
即，对于一个行数为 m，列数为 n，行列下标都从 0 开始编号的二维数组，将其中的每个元素 (i,j) 映射到整数域内，并且它们按照行优先的顺序一一对应着 [0,mn) 中的每一个整数，该映射为：

再将整数 x 映射回其在矩阵中的下标，该映射为：

优化
注意到一维数组的作用是其元素下标（即整数 x）可以对应到二维数组坐标，因此可以利用这个整数直接完成二维数组间的映射，而不再额外使用一个一维数组。
因此，本题求解过程如下：
若两数组元素个数不同，则直接返回；若两数组元素个数相同，则第 x 个元素（x∈[0,mn)）在二维数组中对应的下标为 (x/n,x%n) 。

更多题解

代码

vector 表示二维数组：
1）定义二维数组：vector<vector< int > > v;
2）设置二维数组大小：
a. 指定外层 vector 大小
v.resize(3);
b. 遍历指定内层 vector 大小
for(int i=0;i<3;i++)
v[i].resize(3);
c. 一次指定内外层 vector 大小
v.resize(n, vector< int >(m));

class Solution {
public:
    vector<vector<int>> matrixReshape(vector<vector<int>>& mat, int r, int c) {
        vector<vector<int> > ans(r,vector<int >(c));
        //m :矩阵行	  n :矩阵列
        int m=mat.size(),n=mat[0].size();
        //判断两二维数组元素是否相等
        if(r*c!=m*n){
            return mat;
        }
        //两二维数组间的元素映射
        for(int i=0;i<r*c;++i){
            ans[i/c][i%c]=mat[i/n][i%n];
        }
        return ans;
    }
};