Leetcode_Array -- 566. Reshape the Matrix [easy]

In MATLAB, there is a very useful function called ‘reshape’, which can reshape a matrix into a new one with different size but keep its original data.
You’re given a matrix represented by a two-dimensional array, and two positive integers r and c representing the row number and column number of the wanted reshaped matrix, respectively.

The reshaped matrix need to be filled with all the elements of the original matrix in the same row-traversing order as they were.

If the ‘reshape’ operation with given parameters is possible and legal, output the new reshaped matrix; Otherwise, output the original matrix.

在matlab中有一个非常有用的函数，叫做‘reshape’，它可以将一个矩阵转换成另外一个形状，且保持原来数据不变
给你一个二维数组，和正整数 r 和 c ，分别代表了要重新转换形状矩阵的行和列，
转换形状之后的数组需要以行的顺序填入原始数据
如果reshape函数操作被给定了可能且合法的参数，则返回新的变换后的矩阵，否则返回原使矩阵
Example 1:

Input: 
nums = 
[[1,2],
 [3,4]]
r = 1, c = 4
Output: 
[[1,2,3,4]]
Explanation:
The row-traversing of nums is [1,2,3,4]. The new reshaped matrix is a 1 * 4 matrix, fill it row by row by using the previous list.

Example 2:

Input: 
nums = 
[[1,2],
 [3,4]]
r = 2, c = 4
Output: 
[[1,2],
 [3,4]]
Explanation:
There is no way to reshape a 2 * 2 matrix to a 2 * 4 matrix. So output the original matrix.

Note:
1、The height and width of the given matrix is in range [1, 100].
2、The given r and c are all positive.

Solutions:

Python
(1)
class Solution:
    def matrixReshape(self, nums, r, c):
        """
        :type nums: List[List[int]]
        :type r: int
        :type c: int
        :rtype: List[List[int]]
        """
        if r*c != len(nums)*len(nums[0]):
            return nums
            
        result = []
        temp = []
        count = 0
        for i in range(len(nums)):
            for j in range(len(nums[0])):
                temp.append(nums[i][j])
                count += 1
                if count == c:
                    result.append(temp)
                    temp = []
                    count = 0
        return result

(2)
class Solution:
    def matrixReshape(self, nums, r, c):
        """
        :type nums: List[List[int]]
        :type r: int
        :type c: int
        :rtype: List[List[int]]
        """
        if r*c != len(nums)*len(nums[0]):
            return nums
        numslist = [x for y in nums for x in y]
        result = []
        for i in range(r):
            result.append(numslist[i*c:(i+1)*c])   #a[:A]，这个取值的范围是不包括A的，也就是a数组取值到A-1的位置
        return result

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<vector<int> > result(r);   #这里需要特别注意，二维数组需要初始化，即括号里的r不可少，否则会报错，具体参见参考博客
        int newrow = 0;
        int newcol = 0;
        for (int i = 0;i<row;i++){
            for (int j = 0;j<col;j++){
                result[newrow].push_back(nums[i][j]);
                if((++newcol)%c == 0)++newrow;   #当数据存满一列的时候就换行存储
                    
            }
        }
        return result;

    }
};

参考博客：
关于C++二维数组未初始化报错：
https://blog.csdn.net/m0_38088298/article/details/79249044