566. 重塑矩阵

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

思路：意思就是把一个mn的矩阵变成另一个rc的矩阵，问是否能实现。能肯定矩阵内元素总数一样，然后的话就是所有元素重新赋值，遍历一遍所有元素，然后根据新的列的数量把其放到对应的位置即可。

code

class Solution {
public:
    vector<vector<int>> matrixReshape(vector<vector<int>>& mat, int r, int c) {
    int hang = mat.size();
    int lie  = mat[0].size();
    if(hang * lie != r * c) return mat;
    else{
            vector<vector<int>> ans(r,vector<int>(c));
            for(int i=0;i<hang*lie;i++){
                ans[i/c][i%c] = mat[i/lie][i%lie];
            }
            return ans;
    }
    }
};

复制