5-矩阵专题（杨辉三角+重构矩阵）

记录两道简单题呜呜呜呜

“给定一个非负整数 numRows，生成「杨辉三角」的前 numRows 行。

在「杨辉三角」中，每个数是它左上方和右上方的数的和。”

示例 1:

输入: numRows = 5
输出: [[1],[1,1],[1,2,1],[1,3,3,1],[1,4,6,4,1]]

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/pascals-triangle
 

我想了半天，结果发现就是按照题目的描述用动态规划写代码就可以了...

代码：

class Solution:
    def generate(self, numRows: int) -> List[List[int]]:
        res=[]
        for i in range(numRows):
            row=[]
            for j in range(0,i+1):
                if j==0 or j==i:
                    row.append(1)
                else:
                    x=res[i-1][j-1]+res[i-1][j]
                    row.append(x)
            res.append(row)
        return res

一行行处理，可以看到杨辉三角每一行首位都是1，中间元素就是上一行两元素之和。最后把处理好的每一行合并即可。

另一道：

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

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

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

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

输入：mat = [[1,2],[3,4]], r = 1, c = 4
输出：[[1,2,3,4]]

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/reshape-the-matrix

这里首先需要知道矩阵重构的基本思想：多维->一维->多维

其中，多维矩阵中的元素映射到一维数组（设原多维矩阵是m*n）：

一维数组的元素映射到多维矩阵（设现多维矩阵是r*c）：

根据这些映射关系，就可以轻松写出代码了：

class Solution:
    def matrixReshape(self, mat: List[List[int]], r: int, c: int) -> List[List[int]]:
        m=len(mat)
        n=len(mat[0])
        if r*c!=m*n:
            return mat
        else:
            ans = [[0] * c for _ in range(r)]
        for x in range(m * n):
            ans[x // c][x % c] = mat[x // n][x % n]
        return ans
            # // 是取整除

果然写出代码的前提是将题目描述数学化，学到了！