leetcode hot 100 杨辉三角

118. 杨辉三角

已解答

简单

相关标签

相关企业

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

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

class Solution(object):
    def generate(self, numRows):
        """
        :type numRows: int
        :rtype: List[List[int]]
        """
        ans = []
        if numRows>=1:
            ans.append([1])
        if numRows>=2:
            ans.append([1,1])
        if numRows>=3:
            for i in range(3,numRows+1):
                ans_t=[]
                for j in range(i):
                    if j==0 or j==i-1:
                        ans_t.append(1)
                    else:
                        ans_t.append(ans[-1][j-1]+ans[-1][j])
                ans.append(ans_t)
        return ans