leetcode 566. Reshape the Matrix

朴灿烈づ我的快乐病毒、 2022-06-16 00:22 168阅读 0赞

**1.题目**  
Reshape the Matrix  
例1:  
Input:

nums = [[1,2], [3,4]]
    r = 1, c = 4

Output:

[[1,2,3,4]]

Explanation:  
元素横向展开，2\*2的矩阵转为1\*4的矩阵  
例 2:  
Input:

nums = [[1,2], [3,4]]
    r = 2, c = 4

Output:

[[1,2], [3,4]]

Explanation:  
2\*2的矩阵无法转为2\*4的矩阵，所以输出原矩阵。  
**2.解法**  
（1）origin\_r \* origin\_c 的矩阵 reshape为 r\*c的矩阵，需要满足：  
origin\_r \* origin\_c=r\*c  
（2）元素位置对应的关系  
如果将矩阵横向展开为一维数组，元素个数为n=origin\_r \* origin\_c  
在元素在一维数组中对应的位置i：  
原矩阵位置\[i/origin\_c,i%origin\_c\]  
新矩阵位置\[i/c,i%c\]  
**3.代码**  
c++

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;
        }

python

def matrixReshape(self, nums, r, c):
            origin_r=len(nums)#行
            origin_c=len(nums[0])#列
            if origin_c*origin_r==r*c:#可以reshape
                temp=[num for row in nums for num in row]#展开成一维数组
                newMatrix=[[0 for j in xrange(c)] for i in xrange(r)]
                for i in xrange(r):
                    for j in xrange(c):
                        newMatrix[i][j]=temp[i*c+j]
                return newMatrix
            else:#无法reshape
                return nums