LeetCode 566. 重塑矩阵

题目

题解

• 先存进一维数组，后放入二维数组
• 二维数组的一维表示
  映射处理：二维数组中(i,j)元素是第i×n+j个
  

代码

class Solution {
public:
    vector<vector<int>> matrixReshape(vector<vector<int>>& mat, int r, int c) {
        int m=mat.size();
        int n=mat[0].size();
        if(r*c!=m*n)
            return mat;
        vector<int> data;
        for(int i=0;i<m;i++)
            for(int j=0;j<n;j++)
                data.push_back(mat[i][j]);
        vector<vector<int>> res(r, vector<int>(c, 0));;
        int k=0;
        for(int i=0;i<r;i++)
            for(int j=0;j<c;j++)
            {
                res[i][j]=data[k];
                k++;
            }
        return res;      
    }
};