leetcode_566_重塑矩阵

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

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

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

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

示例 1:

输入: 
nums = 
[[1,2],
 [3,4]]
r = 1, c = 4
输出: 
[[1,2,3,4]]
解释:
行遍历nums的结果是 [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. 给定矩阵的宽和高范围在 [1, 100]。
2. 给定的 r 和 c 都是正数。

---

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

有点慢，注意定义一个数组的方式   vector<vector<int>> result(r, vector<int>(c));

class Solution {
public:
    vector<vector<int>> matrixReshape(vector<vector<int>>& nums, int r, int c) {
         int row = nums.size();
        int col = nums[0].size();
        if (r*c != row*col)
            return nums;
        vector<int> vet;
        for (int i = 0; i < row; i++)
        {
            for (int j = 0; j < col; j++)
                vet.push_back(nums[i][j]);
        }
        vector<vector<int>> vet1;
        int x = 0;
        for (int i = 0; i < r; i++)
        {
           vector<int> vet2;
            for (int j = 0; j < c; j++)
            {
                vet2.push_back(vet[x]);
                x++;
            }
            vet1.push_back(vet2);
        }
        return vet1;
    }
};

该思路是把整个原始的数组变成一维，然后用push_back挨个赋值,

vet1.push_back(vet2) 二维数组挨行赋值

class Solution {
public:

vector<vector<int>> matrixReshape(vector<vector<int>>& nums, int r, int c) {
        //原矩阵的行，列
        int origin_r = nums.size();
        int origin_c = nums[0].size();
        int n = origin_c*origin_r;//元素个数
        if (n == r*c) {
            //新矩阵
            vector<vector<int>> newMaxtrix(r, vector<int>(c, 0));
            for (int i = 0; i < n; i++)
                newMaxtrix[i / c][i%c] = nums[i / origin_c][i%origin_c];
            return newMaxtrix;
        }
        else
            return nums;
    }
};

注意0/4=0 0%4=0 1/4=0 1%4=1