leetcode 566. Reshape the Matrix

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