力扣566 重塑矩阵

力扣566 重塑矩阵

题干

在MATLAB中，有一个非常有用的函数 reshape，它可以将一个矩阵重塑为另一个大小不同的新矩阵，但保留其原始数据。

给出一个由二维数组表示的矩阵，以及两个正整数r和c，分别表示想要的重构的矩阵的行数和列数。

重构后的矩阵需要将原始矩阵的所有元素以相同的行遍历顺序填充。

如果具有给定参数的reshape操作是可行且合理的，则输出新的重塑矩阵；否则，输出原始矩阵。

示例

输入: nums = [[1,2], [3,4]] r = 1, c = 4 输出: [[1,2,3,4]] 解释:
行遍历nums的结果是 [1,2,3,4]。新的矩阵是 1 * 4 矩阵, 用之前的元素值一行一行填充新矩阵。

输入: nums = [[1,2], [3,4]] r = 2, c = 4 输出: [[1,2], [3,4]] 解释:
没有办法将 2 * 2 矩阵转化为 2 * 4 矩阵。 所以输出原矩阵。

解析

本题看起来很复杂，好像没有什么比较精巧的方法，貌似只能用暴力法破解，本人也是使用暴力法的，但是在使用暴力法的同时，也要注意整理思路和每一步的方法。

1、暴力法总体思路

1、不能以给定的 r 和 c进行重构，就像示例2一样的情况，直接返回原数组nums。
2、nums可以根据 r 和 c 进行重构，此时需要将原数组nums转变为一个一维数组temp，元素按顺序排列。
3、根据 r 和 c 的值，将一维数组temp中的元素按顺序（或按规律）存放到结果数组results中。

这里就提出了本题需要注意的两个点：
1、如何将二维数组转变为一维数组
2、如何将一维数组转变为二维数组

2、将二维数组转变为一维数组

这里列举出本人自己写的和力扣题解中看到的方法

1、列表加法
首先是本人写的，相对比较丑陋。采用的是列表“+”操作，列表间使用加法运算就是列表拼接，通过把每一行的小列表拼接，形成一维数组。

temp = list()
for num in nums:
	temp += num

2、列表表达式
通过列表表达式生成列表，注意使用了两个for循环，个人觉得从写法上比较美观。

temp = [num for row_num in nums for num in row_num]

3、将一维数组转变为二维数组

1、切片
本人使用的是切片法，直接根据切片从temp中将结果依次“切到”results中返回。同样，看着非常丑陋。

results = list()
for row in range(r):
    results.append([])
    for col in range(c):
        results[row] = temp[row*c : (row+1)*c]

2、数学法（取商取余）
该方法的实现在于，一维数组temp的索引值 i 与二维数组results的行号列号间的关系。
不难发现，其实 i // c 就是temp[i]在results中所在行的行号，而 i % c 就是temp[i]在results中所在列的列号，知道这一点之后，就可以写出比较美观的代码了

results = [[0 for col in range(c)] for row in range(r)]
for i in range(len(temp)):
	results[i//c][i%c] = temp[i]

4、完整代码

class Solution:
    def matrixReshape(self, nums: List[List[int]], r: int, c: int) -> List[List[int]]:
        if len(nums) * len(nums[0]) != r * c:
            return nums
        # 拿到一维数组
        temp = [num for row_num in nums for num in row_num]
        # 映射到二位数组
        results = [[0 for col in range(c)] for row in range(r)]
        for i in range(len(temp)):
            results[i//c][i%c] = temp[i]
        return results

小结

同样是一道简单题中比较简单的题，能不能以美观的方式书写代码也很重要。同样的解题思路，具体代码使用的方法可以体现一个人的知识储备。