一、題目介紹

在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, 100]。
給定的 r 和 c 都是正數。

來源：力扣（LeetCode）
連結：https://leetcode-cn.com/problems/reshape-the-matrix
著作權歸領釦網路所有。商業轉載請聯絡官方授權，非商業轉載請註明出處。

二、解題思路

本題主要考察對二維陣列的座標計算。具體實現請見程式碼。

三、解題程式碼

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;
        else
        {
            vector<vector<int>> vec(r, vector<int>(c, 0));
            for(int i = 0; i < m; ++i)
            {
                for(int j = 0; j < n; ++j)
                {
                    int x = (i * n + j) / c;
                    int y = (i * n + j) % c;
                    vec[x][y] = nums[i][j];
                }
            }
            return vec;
        }
    }
};

四、解題結果