1. 程式人生 > >Leetcode566.Reshape the Matrix重塑矩陣

Leetcode566.Reshape the Matrix重塑矩陣

在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) {
        int R = nums.size();
        if(R == 0)
            return nums;
        int C = nums[0].size();
        if(r * c != R * C)
            return nums;
        vector<vector<int> >res;
        int cntR = 0;
        int cntC = 0;
        for(int i = 0; i < r; i++)
        {
            vector<int> temp;
            for(int j = 0; j < c; j++)
            {
                temp.push_back(nums[cntR][cntC]);
                if(cntC == C - 1)
                {
                    cntC = 0;
                    cntR++;
                }
                else
                {
                    cntC++;
                }
            }
            res.push_back(temp);
        }
        return res;
    }
};