leetcode--566. 重塑矩阵

给出一个由二维数组表示的矩阵，以及两个正整数r和c，分别表示想要的重构的矩阵的行数和列数。

重构后的矩阵需要将原始矩阵的所有元素以相同的行遍历顺序填充。

如果具有给定参数的reshape操作是可行且合理的，则输出新的重塑矩阵；否则，输出原始矩阵。

/*
 * 输入:
nums =
[[1,2],
 [3,4]]
r = 1, c = 4
输出:
[[1,2,3,4]]
解释:
行遍历nums的结果是 [1,2,3,4]。新的矩阵是 1 * 4 矩阵, 用之前的元素值一行一行填充新矩阵。

 * */
#include <iostream>
#include <vector>
using namespace std;
class Solution {
public:
    vector<vector<int>> matrixReshape(vector<vector<int>>& nums, int r, int c) {
        int m=nums.size(); //行数
        int n=nums[0].size();//列数
        if(m*n != r*c) return nums;

        vector<vector<int>> res;
        vector<int> temp;
        int row=0;//新数组的行
        int cols=0;//新数组的列
        for(int i=0;i<m;i++){

            for(int j=0;j<n;j++){
                temp.push_back(nums[i][j]);
                cols++;
                if(cols==c){
                    res.push_back(temp);
                    temp.clear();
                    cols=0;
                }
            }
        }
        for(int i=0;i<r;i++){

            for(int j=0;j<c;j++){
               cout<<res[i][j]<<" ";
            }
            cout<<endl;
        }
        return res;
    }

};

int main() {
    vector<vector<int>> nums={{1,2},{3,4}};
    Solution p;
    p.matrixReshape(nums,2,4);
    return 0;
}