LeetCode 118. Pascal's Triangle

题目

思路

用一个list记录上一层的数字，用来更新当前层。

代码

class Solution:
    def generate(self, numRows):
        """
        :type numRows: int
        :rtype: List[List[int]]
        """
        res_list = []
        last_list = []
        for i in range(1, numRows + 1):
            tmp_list = []
            for j in range(i):
                if j == 0: tmp_list.append(1)
                elif j == i - 1: tmp_list.append(1)
                else: tmp_list.append(last_list[j - 1] + last_list[j])
            last_list[:] = tmp_list[:]
            res_list.append(tmp_list[:])
        return res_list