leetcode#118. 杨辉三角

题目描述

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

题目地址：
https://leetcode-cn.com/problems/pascals-triangle/

要点分析

1. 每一行的头和尾元素均初始化为1；
2. 中间元素由上层[i-1][j-1]+[i-1][j]构成；
3. 运用动态规划思想通过前层生成后层，构造一个ans=[ ]用来存储结果，每一层的结果append到ans中，numRows=0时不执行循环直接返回ans=[ ]；
4. 每行的长度可以预先确定。

class Solution:
    def generate(self, numRows: int) -> List[List[int]]:
        ans = []
        for i in range(numRows):
            row=[0] * (i+1)
            # row = [0 for _ in range(i+1)]
            # 已经根据杨辉三角特点确定每行的长度
            row[0],row[-1] = 1, 1
            
            for j in range(1,i):
                row[j] = ans[i-1][j-1]+ans[i-1][j]
                
            ans.append(row)
        
        return ans

优化至空间复杂度O(K)

class Solution:
    def getRow(self, rowIndex: int) -> List[int]:
        ans = [1]*(rowIndex+1)
        for i in range(rowIndex+1):
            for j in range(i,1,-1):
                ans[j-1] = ans[j-2] + ans[j-1]

        return ans