leetcode_118_119_杨辉三角

杨辉三角

描述

简单

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

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

示例:

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

解题

相当于把三角的数据进行左对齐，得

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

除第1、2行和每行的左右两边的数据外，其余的数据等于左上方的数据加上上方的数据

class Solution:
    def generate(self, numRows: int) -> List[List[int]]:
        if numRows == 0:
            return []
        elif numRows == 1:
            return [[1]]
        elif numRows == 2:
            return [[1], [1, 1]]
        
        trangle = [[1], [1, 1]]

        for i in range(2, numRows):
            temp = [1 for _ in range(i+1)]
            # 左右两边的数据不进行计算
            for j in range(1, len(temp)-1):
                temp[j] = trangle[i-1][j-1] + trangle[i-1][j]

            trangle.append(temp)
        
        return trangle

杨辉三角II

描述

简单

给定一个非负索引 k，其中 k ≤ 33，返回杨辉三角的第 k 行。

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

示例:

输入: 3
输出: [1,3,3,1]

进阶：

你可以优化你的算法到 O(k) 空间复杂度吗？

解题

按照上面的解法迭代到k层再返回也是可以的，但空间复杂度就不能满足要求了

首先申请所需的空间，也是多次迭代更新

每次更新时从后往前更新，这样就不会影响下一个数的更新

class Solution:
    def getRow(self, rowIndex: int) -> List[int]:
        row = [1 for i in range(rowIndex+1)]
        for i in range(rowIndex+1):
            # 每次每层左右两端的数都不更新
            for j in range(i-1, 0, -1):
                row[j] = row[j] + row[j-1]
        return row