118. 楊輝三角

給定一個非負整數 numRows，生成楊輝三角的前 numRows 行。

在楊輝三角中，每一個數是它左上方和右上方的數的和。

 

 

 1 class Solution(object):
 2     def generate(self, numRows):
 3         """
 4         :type numRows: int
 5         :rtype: List[List[int]]
 6         """
 7         a = []
 8         if numRows == 0:
 9             return a
10         # 初始化
11         for i in range(1, numRows + 1):
12             a.append([0] * i)
13         print(a)
14         # 最頂端元素
15         a[0][0] = 1
16         for i in range(1, numRows):
17             for j in range(i + 1):
18                 # 每一行第一個和最後一個元素都是1
19                 if j == i or j == 0:
20                     a[i][j] = 1
21                 # 不然是肩上兩個元素之和
22                 else:
23                     a[i][j] = a[i - 1][j] + a[i - 1][j - 1]
24         return a
25 
26 
27 if __name__ == '__main__':
28     solution = Solution()
29     print(solution.generate(5))